Fix test_warnings on Windows (don't change current directory)

Victor Stinner · Victor Stinner · commit d656354dddf3 · 2010-05-20T21:42:00.000Z
diff --git a/Lib/test/test_warnings.py b/Lib/test/test_warnings.py
@@ -4,7 +4,6 @@
 from io import StringIO
 import sys
 import unittest
-import shutil
 import tempfile
 import subprocess
 from test import support
@@ -678,12 +677,11 @@ def test_issue_8766(self):
         # "import encodings" emits a warning whereas the warnings is not loaded
         # or not completly loaded (warnings imports indirectly encodings by
         # importing linecache) yet
-        old_cwd = os.getcwd()
+        cwd = tempfile.mkdtemp()
         try:
-            cwd = tempfile.mkdtemp()
+            encodings = os.path.join(cwd, 'encodings')
+            os.mkdir(encodings)
             try:
-                os.chdir(cwd)
-                os.mkdir('encodings')
                 env = os.environ.copy()
                 env['PYTHONPATH'] = cwd
 
@@ -697,9 +695,9 @@ def test_issue_8766(self):
                     env=env)
                 self.assertEqual(retcode, 0)
             finally:
-                shutil.rmtree(cwd)
+                os.rmdir(encodings)
         finally:
-            os.chdir(old_cwd)
+            os.rmdir(cwd)
 
 def test_main():
     py_warnings.onceregistry.clear()
