Merge pull request #820 from wwwjfy/w606-tweak

refs W606, only NAME can follow await

2 files changed, 5 insertions, 1 deletions

diff --git a/pycodestyle.py b/pycodestyle.py
index c9e1667..73f4076 100755
--- a/pycodestyle.py
+++ b/pycodestyle.py
@@ -1580,7 +1580,7 @@ def python_3000_async_await_keywords(logical_line, tokens):
             else:
                 error = True
         elif state[0] == 'await':
-            if token_type in (tokenize.NAME, tokenize.NUMBER, tokenize.STRING):
+            if token_type == tokenize.NAME:
                 # An await expression. Return to looking for async/await
                 # names.
                 state = None
diff --git a/testsuite/W60.py b/testsuite/W60.py
index 4cbaad9..78b163e 100644
--- a/testsuite/W60.py
+++ b/testsuite/W60.py
@@ -45,6 +45,10 @@ async = 42
 #: W606
 await = 42
 #: W606
+await 42
+#: W606
+await 'test'
+#: W606
 def async():
     pass
 #: W606