Avoid removing whitespace for walrus operators within subscripts (#3823)

[etc/vim.git] / tests / data / py_38 / pep_572_remove_parens.py

diff --git a/tests/data/py_38/pep_572_remove_parens.py b/tests/data/py_38/pep_572_remove_parens.py
index 9718d95b499de4e2c966de31d10433deb471e681..b952b2940c502f87bf8dabfc7ba02d2e159a8e04 100644 (file)
--- a/tests/data/py_38/pep_572_remove_parens.py
+++ b/tests/data/py_38/pep_572_remove_parens.py
@@ -49,6 +49,25 @@ def a():
 def this_is_so_dumb() -> (please := no):
     pass
 
 def this_is_so_dumb() -> (please := no):
     pass
 

+async def await_the_walrus():
+    with (x := y):
+        pass
+
+    with (x := y) as z, (a := b) as c:
+        pass
+
+    with (x := await y):
+        pass
+
+    with (x := await a, y := await b):
+        pass
+
+    with ((x := await a, y := await b)):
+        pass
+
+    with (x := await a), (y := await b):
+        pass
+

 
 # output
 if foo := 0:
 
 # output
 if foo := 0:


@@ -103,3 +122,22 @@ def a():
 def this_is_so_dumb() -> (please := no):
     pass
 
 def this_is_so_dumb() -> (please := no):
     pass
 

+
+async def await_the_walrus():
+    with (x := y):
+        pass
+
+    with (x := y) as z, (a := b) as c:
+        pass
+
+    with (x := await y):
+        pass
+
+    with (x := await a, y := await b):
+        pass
+
+    with (x := await a, y := await b):
+        pass
+
+    with (x := await a), (y := await b):
+        pass