bpo-34136: Make test_do_not_recreate_annotations more reliable. (GH-8364)

serhiy-storchaka · web-flow · commit 06ca3f0c09d0 · 2018-07-21T07:44:04.000+03:00
diff --git a/Lib/test/test_opcodes.py b/Lib/test/test_opcodes.py
@@ -1,7 +1,7 @@
 # Python test set -- part 2, opcodes
 
 import unittest
-from test import ann_module
+from test import ann_module, support
 
 class OpcodeTest(unittest.TestCase):
 
@@ -42,10 +42,14 @@ def test_use_existing_annotations(self):
         self.assertEqual(ns['__annotations__'], {'x': int, 1: 2})
 
     def test_do_not_recreate_annotations(self):
-        class C:
-            del __annotations__
-            with self.assertRaises(NameError):
-                x: int
+        annotations = {}
+        # Don't rely on the existence of the '__annotations__' global.
+        with support.swap_item(globals(), '__annotations__', annotations):
+            class C:
+                del __annotations__
+                x: int  # Updates the '__annotations__' global.
+        self.assertIn('x', annotations)
+        self.assertIs(annotations['x'], int)
 
     def test_raise_class_exceptions(self):
 
