Fix for list_comp, add tests

krshrimali · Jul 17, 2022 · 1fac76b · 1fac76b
1 parent afc4804
commit 1fac76b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/lpython/parser/parser.yy b/src/lpython/parser/parser.yy
@@ -679,6 +679,7 @@ id_list
 id_item
     : id_list { $$ = ID_TUPLE_01($1, @$); }
     | id_list "," { $$ = ID_TUPLE_03($1, @$); }
+    | "(" id ")" { $$ = $2; }
     | "(" id_list "," ")" { $$ = ID_TUPLE_03($2, @$); }
     | "(" id_list ","  id ")" { $$ = ID_TUPLE_01(TUPLE_($2, $4), @$); }
     ;

diff --git a/tests/parser/comprehension1.py b/tests/parser/comprehension1.py
@@ -11,3 +11,7 @@
 distance_orig = [x**2 + y**2 + z**2 for x, y, z in points]
 
 odd_elements = [i for i in main_list if i & 1]
+
+first_ten_elements = [i for (i) in range(10)]
+
+another_ten_elements = [(i) for (i) in range(10)]