Fix tests

psf · May 7, 2021 · 3c92cc3 · 3c92cc3
1 parent 9df6c76
commit 3c92cc3
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 27 deletions.
diff --git a/tests/data/empty_lines.py b/tests/data/empty_lines.py
@@ -123,23 +123,31 @@ def f():
             return NO
 
         if prevp.type == token.EQUAL:
-            if prevp.parent and prevp.parent.type in {
-                syms.typedargslist,
-                syms.varargslist,
-                syms.parameters,
-                syms.arglist,
-                syms.argument,
-            }:
+            if (
+                prevp.parent
+                and prevp.parent.type
+                in {
+                    syms.typedargslist,
+                    syms.varargslist,
+                    syms.parameters,
+                    syms.arglist,
+                    syms.argument,
+                }
+            ):
                 return NO
 
         elif prevp.type == token.DOUBLESTAR:
-            if prevp.parent and prevp.parent.type in {
-                syms.typedargslist,
-                syms.varargslist,
-                syms.parameters,
-                syms.arglist,
-                syms.dictsetmaker,
-            }:
+            if (
+                prevp.parent
+                and prevp.parent.type
+                in {
+                    syms.typedargslist,
+                    syms.varargslist,
+                    syms.parameters,
+                    syms.arglist,
+                    syms.dictsetmaker,
+                }
+            ):
                 return NO
 
 
@@ -177,11 +185,15 @@ def g():
             return NO
 
         if prevp.type == token.EQUAL:
-            if prevp.parent and prevp.parent.type in {
-                syms.typedargslist,
-                syms.varargslist,
-                syms.parameters,
-                syms.arglist,
-                syms.argument,
-            }:
+            if (
+                prevp.parent
+                and prevp.parent.type
+                in {
+                    syms.typedargslist,
+                    syms.varargslist,
+                    syms.parameters,
+                    syms.arglist,
+                    syms.argument,
+                }
+            ):
                 return NO
diff --git a/tests/data/expression.diff b/tests/data/expression.diff
@@ -272,8 +272,8 @@
 -for addr_family, addr_type, addr_proto, addr_canonname, addr_sockaddr in socket.getaddrinfo('google.com', 'http'):
 +print(*lambda x: x)
 +assert not Test, "Short message"
-+assert this is ComplexTest and not requirements.fit_in_a_single_line(
-+    force=False
++assert (
++    this is ComplexTest and not requirements.fit_in_a_single_line(force=False)
 +), "Short message"
 +assert parens is TooMany
 +for (x,) in (1,), (2,), (3,):

diff --git a/tests/data/expression.py b/tests/data/expression.py
@@ -510,8 +510,8 @@ async def f():
 print(**{1: 3} if False else {x: x for x in range(3)})
 print(*lambda x: x)
 assert not Test, "Short message"
-assert this is ComplexTest and not requirements.fit_in_a_single_line(
-    force=False
+assert (
+    this is ComplexTest and not requirements.fit_in_a_single_line(force=False)
 ), "Short message"
 assert parens is TooMany
 for (x,) in (1,), (2,), (3,):

diff --git a/tests/data/expression_skip_magic_trailing_comma.diff b/tests/data/expression_skip_magic_trailing_comma.diff
@@ -253,8 +253,8 @@
 -for addr_family, addr_type, addr_proto, addr_canonname, addr_sockaddr in socket.getaddrinfo('google.com', 'http'):
 +print(*lambda x: x)
 +assert not Test, "Short message"
-+assert this is ComplexTest and not requirements.fit_in_a_single_line(
-+    force=False
++assert (
++    this is ComplexTest and not requirements.fit_in_a_single_line(force=False)
 +), "Short message"
 +assert parens is TooMany
 +for (x,) in (1,), (2,), (3,):