diff --git a/Lib/test/test_compileall.py b/Lib/test/test_compileall.py --- a/Lib/test/test_compileall.py +++ b/Lib/test/test_compileall.py @@ -98,29 +98,18 @@ class CompileallTests(unittest.TestCase) # Test against bad files self.add_bad_source_file() self.assertFalse(compileall.compile_file(self.bad_source_path, force=False, quiet=2)) self.assertFalse(compileall.compile_dir(self.directory, 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() self.assertFalse(compileall.compile_path(skip_curdir=False, force=True, quiet=2)) def test_no_pycache_in_non_package(self): # Bug 8563 reported that __pycache__ directories got created by