clean the environment from pre-existing PYTHONWARNINGS for test_warnings

ambv · ambv · commit 6ae466732c6a · 2013-04-24T01:49:52.000+02:00
diff --git a/Lib/unittest/test/test_runner.py b/Lib/unittest/test/test_runner.py
@@ -139,6 +139,18 @@ def cleanup2():
 class Test_TextTestRunner(unittest.TestCase):
     """Tests for TextTestRunner."""
 
+    def setUp(self):
+        # clean the environment from pre-existing PYTHONWARNINGS to make
+        # test_warnings results consistent
+        self.pythonwarnings = os.environ.get('PYTHONWARNINGS')
+        if self.pythonwarnings:
+            del os.environ['PYTHONWARNINGS']
+
+    def tearDown(self):
+        # bring back pre-existing PYTHONWARNINGS if present
+        if self.pythonwarnings:
+            os.environ['PYTHONWARNINGS'] = self.pythonwarnings
+
     def test_init(self):
         runner = unittest.TextTestRunner()
         self.assertFalse(runner.failfast)
