Merge pull request #1032 from cdce8p/walrus-multiple-spaces

Emit E221-E224 for walrus op

Author: asottile

pycodestyle.py

@@ -152,7 +152,7 @@ def lru_cache(maxsize=128):  # noqa as it's a fake implementation.
 COMPARE_TYPE_REGEX = re.compile(r'(?:[=!]=|is(?:\s+not)?)\s+type(?:s.\w+Type'
                                 r'|\s*\(\s*([^)]*[^ )])\s*\))')
 KEYWORD_REGEX = re.compile(r'(\s*)\b(?:%s)\b(\s*)' % r'|'.join(KEYWORDS))
-OPERATOR_REGEX = re.compile(r'(?:[^,\s])(\s*)(?:[-+*/|!<=>%&^]+)(\s*)')
+OPERATOR_REGEX = re.compile(r'(?:[^,\s])(\s*)(?:[-+*/|!<=>%&^]+|:=)(\s*)')
 LAMBDA_REGEX = re.compile(r'\blambda\b')
 HUNK_REGEX = re.compile(r'^@@ -\d+(?:,\d+)? \+(\d+)(?:,(\d+))? @@.*$')
 STARTSWITH_DEF_REGEX = re.compile(r'^(async\s+def|def)\b')

testsuite/python38.py

@@ -42,3 +42,14 @@ def f3(
 all_the_things: t.List[str] = []
 
 import logging
+#: E221:1:5 E222:1:9 E221:3:6
+if x  :=  1:
+    pass
+if (x  := 2):
+    pass
+#: E223:1:5 E224:1:8
+if x	:=	2:
+    pass
+#: E221:1:6 E221:1:19
+if (x  := 1) == (y  := 2):
+    pass