This test failed under python -O.

rewrite_file(): Delete both .pyc and .pyo leftovers, and explicitly close the new source file after writing to it.
skrah · Aug 2, 2001 · d342c62 · d342c62
1 parent 044bb4d
commit d342c62
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/Lib/test/test_pkgimport.py b/Lib/test/test_pkgimport.py
@@ -36,9 +36,13 @@ def tearDown(self):
         self.remove_modules()
 
     def rewrite_file(self, contents):
-        compiled_path = self.module_path + 'c'
-        if os.path.exists(compiled_path): os.remove(compiled_path)
-        open(self.module_path, 'w').write(contents)
+        for extension in "co":
+            compiled_path = self.module_path + extension
+            if os.path.exists(compiled_path):
+                os.remove(compiled_path)
+        f = open(self.module_path, 'w')
+        f.write(contents)
+        f.close()
 
     def test_package_import__semantics(self):