From: Berker Peksag <berker.peksag@gmail.com>
Date: Wed, 28 Sep 2016 14:38:53 +0000 (+0300)
Subject: Issue #27322: Set sys.path to a temp dir in test_compile_path
X-Git-Tag: v3.6.0b2~99
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=408b78c06f73ad2546e6fcb93e6dcef7d248f4a1;p=python

Issue #27322: Set sys.path to a temp dir in test_compile_path
---

diff --git a/Lib/test/test_compileall.py b/Lib/test/test_compileall.py
index 9b424a7250..1f05e78c23 100644
--- a/Lib/test/test_compileall.py
+++ b/Lib/test/test_compileall.py
@@ -103,19 +103,8 @@ class CompileallTests(unittest.TestCase):
                                                 force=False, quiet=2))
 
     def test_compile_path(self):
-        # Exclude Lib/test/ which contains invalid Python files like
-        # Lib/test/badsyntax_pep3120.py
-        testdir = os.path.realpath(os.path.dirname(__file__))
-        if testdir in sys.path:
-            self.addCleanup(setattr, sys, 'path', sys.path)
-
-            sys.path = list(sys.path)
-            try:
-                sys.path.remove(testdir)
-            except ValueError:
-                pass
-
-        self.assertTrue(compileall.compile_path(quiet=2))
+        with test.test_importlib.util.import_state(path=[self.directory]):
+            self.assertTrue(compileall.compile_path(quiet=2))
 
         with test.test_importlib.util.import_state(path=[self.directory]):
             self.add_bad_source_file()