bpo-28655: Fix test_import.test_missing_source_legacy() (GH-9589)

vstinner · ncoghlan · web-flow · commit 43500a5907eb · 2018-09-26T09:09:32.000-07:00
bpo-28655, bpo-33053: test_import.test_missing_source_legacy() now removes the .pyc file that it creates to avoid leaking a file. Fix extract from commit d5d9e02. Co-Authored-By: Nick Coghlan <ncoghlan@gmail.com>
diff --git a/Lib/test/test_import/__init__.py b/Lib/test/test_import/__init__.py
@@ -779,8 +779,11 @@ def test_missing_source_legacy(self):
         unload(TESTFN)
         importlib.invalidate_caches()
         m = __import__(TESTFN)
-        self.assertEqual(m.__file__,
-                         os.path.join(os.curdir, os.path.relpath(pyc_file)))
+        try:
+            self.assertEqual(m.__file__,
+                             os.path.join(os.curdir, os.path.relpath(pyc_file)))
+        finally:
+            os.remove(pyc_file)
 
     def test___cached__(self):
         # Modules now also have an __cached__ that points to the pyc file.
