Update unit tests

Author: jaepil

tests/expressions/test_interpreter_coverage1.py

@@ -608,7 +608,9 @@ def test_global_name_error():
 """
     # 'x' is declared global but never defined in global_env => NameError
     tree = parser.parse(code)
-    with pytest.raises(NameError, match="Global name 'x' is not defined"):
+    with pytest.raises(
+        SyntaxError, match="global declaration not allowed at module level"
+    ):
         interpreter.execute(tree)
 
 
@@ -1308,6 +1310,39 @@ def inner():
     with pytest.raises(SyntaxError, match="name 'x' is nonlocal and global"):
         interpreter.execute(tree)
 
+    # Test global declaration in nested function
+    code = """
+x = 1
+def outer():
+    x = 2
+    def inner():
+        global x
+        return x
+    return inner()
+
+result = outer()
+"""
+    tree = parser.parse(code)
+    interpreter.execute(tree)
+    assert interpreter.get_name_value("result") == 1
+
+    # Test global declaration
+    code = """
+x = 1
+def foo():
+    global x
+    x = 2
+    return x
+
+result1 = x
+foo()
+result2 = x
+"""
+    tree = parser.parse(code)
+    interpreter.execute(tree)
+    assert interpreter.get_name_value("result1") == 1
+    assert interpreter.get_name_value("result2") == 2
+
 
 def test_visit_nonlocal():
     parser = ExpressionsParser()

tests/expressions/test_interpreter_generator.py

@@ -33,14 +33,10 @@ def subgen():
     return 'done'
 
 def gen():
-    print("script: GEN")
     output = yield from subgen()
-    print("script: OUTPUT", output)
     yield output
 
-print(gen)
 result = list(gen())
-print(result)
 """
     parser = ExpressionsParser()
     interpreter = ExpressionsInterpreter()
@@ -111,7 +107,6 @@ def gen():
     return gen
 
 g = make_gen()
-print(g)
 result1 = list(g())
 result2 = list(g())
 """

tests/expressions/test_interpreter_stmt_with.py

@@ -67,7 +67,7 @@ def test_with_statement_scope_isolation(self, parser, interpreter):
         with pytest.raises(NameError):
             interpreter.execute(tree)
 
-        assert "outer" in interpreter.local_env
+        assert "outer" in interpreter.global_env
         assert "inner" not in interpreter.local_env
 
     def test_with_statement_global_access(self, parser, interpreter):
@@ -82,7 +82,6 @@ def test_with_statement_global_access(self, parser, interpreter):
         interpreter.global_env["cm"] = cm
 
         code = """
-global x
 x = 'global'
 with cm:
     global x
@@ -230,8 +229,8 @@ def test_with_statement_variable_lifetime(self, parser, interpreter):
         tree = parser.parse(code)
         interpreter.execute(tree)
 
-        assert "before" in interpreter.local_env
-        assert "after" in interpreter.local_env
+        assert "before" in interpreter.global_env
+        assert "after" in interpreter.global_env
         assert "during" not in interpreter.local_env
         assert interpreter.get_name_value("before") == "modified"