RustPython
diff --git a/‎Lib/test/test_global.py
Copy file name to clipboardExpand all lines: Lib/test/test_global.py
+2-9Lines changed: 2 additions & 9 deletions b/‎Lib/test/test_global.py
Copy file name to clipboardExpand all lines: Lib/test/test_global.py
+2-9Lines changed: 2 additions & 9 deletions
@@ -9,14 +9,9 @@
 class GlobalTests(unittest.TestCase):
 
     def setUp(self):
-        self._warnings_manager = check_warnings()
-        self._warnings_manager.__enter__()
+        self.enterContext(check_warnings())
         warnings.filterwarnings("error", module="<test string>")
 
-    def tearDown(self):
-        self._warnings_manager.__exit__(None, None, None)
-
-
     def test1(self):
         prog_text_1 = """\
 def wrong1():
@@ -54,9 +49,7 @@ def test4(self):
 
 
 def setUpModule():
-    cm = warnings.catch_warnings()
-    cm.__enter__()
-    unittest.addModuleCleanup(cm.__exit__, None, None, None)
+    unittest.enterModuleContext(warnings.catch_warnings())
     warnings.filterwarnings("error", module="<test string>")