add unittest for generator early binding or free vars in outermost loop

Author: cosminbasca

graalpython/com.oracle.graal.python.test/src/tests/test_generators.py

@@ -168,11 +168,3 @@ def f():
 
 if sys.version_info.minor == 4 and sys.version_info.micro < 3:
     del ExceptionTest
-
-
-def test_unboundlocalerror_gen():
-    def func(args):
-        args = (arg for arg in args)
-        return set(args)
-
-    assert func([1, 2, 3, 4]) == {1, 2, 3, 4}

graalpython/com.oracle.graal.python.test/src/tests/test_scope.py

@@ -850,3 +850,24 @@ def register():
         register()
 
     assert MyClass.a_counter == 24
+
+
+def test_generator_scope():
+    my_obj = [1, 2, 3, 4]
+    my_obj = (i for i in my_obj for j in y)
+    y = [1, 2]
+
+    assert set(my_obj.gi_code.co_cellvars) == set()
+    assert set(my_obj.gi_code.co_freevars) == {'y'}
+
+
+def test_func_scope():
+    my_obj = [1, 2, 3, 4]
+
+    def my_obj():
+        return [i for i in my_obj for j in y]
+
+    y = [1, 2]
+
+    assert set(my_obj.__code__.co_cellvars) == set()
+    assert set(my_obj.__code__.co_freevars) == {'my_obj', 'y'}