Merge pull request #820 from wwwjfy/w606-tweak

refs W606, only NAME can follow await

Author: sigmavirus24

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

testsuite/W60.py

@@ -45,6 +45,10 @@
 #: W606
 await = 42
 #: W606
+await 42
+#: W606
+await 'test'
+#: W606
 def async():
     pass
 #: W606