Merge pull request RustPython#2499 from fanninpm/test-global

coolreader18 · web-flow · commit c8fc9e08364e · 2021-02-21T17:51:41.000-06:00
Add test_global from CPython 3.8.7
diff --git a/Lib/test/test_global.py b/Lib/test/test_global.py
@@ -0,0 +1,61 @@
+"""Verify that warnings are issued for global statements following use."""
+
+from test.support import run_unittest, check_syntax_error, check_warnings
+import unittest
+import warnings
+
+
+class GlobalTests(unittest.TestCase):
+
+    def setUp(self):
+        self._warnings_manager = check_warnings()
+        self._warnings_manager.__enter__()
+        warnings.filterwarnings("error", module="<test string>")
+
+    def tearDown(self):
+        self._warnings_manager.__exit__(None, None, None)
+
+
+    def test1(self):
+        prog_text_1 = """\
+def wrong1():
+    a = 1
+    b = 2
+    global a
+    global b
+"""
+        check_syntax_error(self, prog_text_1, lineno=4, offset=5)
+
+    def test2(self):
+        prog_text_2 = """\
+def wrong2():
+    print(x)
+    global x
+"""
+        check_syntax_error(self, prog_text_2, lineno=3, offset=5)
+
+    def test3(self):
+        prog_text_3 = """\
+def wrong3():
+    print(x)
+    x = 2
+    global x
+"""
+        check_syntax_error(self, prog_text_3, lineno=4, offset=5)
+
+    def test4(self):
+        prog_text_4 = """\
+global x
+x = 2
+"""
+        # this should work
+        compile(prog_text_4, "<test string>", "exec")
+
+
+def test_main():
+    with warnings.catch_warnings():
+        warnings.filterwarnings("error", module="<test string>")
+        run_unittest(GlobalTests)
+
+if __name__ == "__main__":
+    test_main()
