Improve async with for syntax

evhub · May 28, 2023 · 9d7d11c · 9d7d11c
1 parent aaf6925
commit 9d7d11c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/coconut/compiler/grammar.py b/coconut/compiler/grammar.py
@@ -2132,8 +2132,8 @@ class Grammar(object):
         | labeled_group(
             (any_len_perm(
                 keyword("match"),
-                required=(keyword("async"),)
-            ) + keyword("with") + keyword("for")).suppress()
+                required=(keyword("async"), keyword("with")),
+            ) + keyword("for")).suppress()
             + many_match + keyword("in").suppress()
             - test
             - suite_with_else_tokens,

diff --git a/coconut/tests/src/cocotest/target_35/py35_test.coco b/coconut/tests/src/cocotest/target_35/py35_test.coco
@@ -68,6 +68,12 @@ def py35_test() -> bool:
             got.append(i)
         assert got == range(5) |> list
 
+        # with for match syntax 4
+        got = []
+        async with match for int(i) in aclosing(arange(5)):
+            got.append(i)
+        assert got == range(5) |> list
+
         return True
     loop.run_until_complete(afor_test())