Issue #27322: Merge from 3.6

berkerpeksag · berkerpeksag · commit e075537432e8 · 2016-09-28T17:39:20.000+03:00
diff --git a/Lib/test/test_compileall.py b/Lib/test/test_compileall.py
@@ -103,19 +103,8 @@ def test_compile_files(self):
                                                 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()
