make file closing more robust

skrah · Jun 13, 2009 · 16a1f63 · 16a1f63
1 parent 94f5583
commit 16a1f63
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/Lib/test/test_import.py b/Lib/test/test_import.py
@@ -46,13 +46,12 @@ def test_with_extension(ext):
             else:
                 pyc = TESTFN + ".pyc"
 
-            f = open(source, "w")
-            print("# This tests Python's ability to import a", ext, "file.", file=f)
-            a = random.randrange(1000)
-            b = random.randrange(1000)
-            print("a =", a, file=f)
-            print("b =", b, file=f)
-            f.close()
+            with open(source, "w") as f:
+                print("# This tests Python's ability to import a", ext, "file.", file=f)
+                a = random.randrange(1000)
+                b = random.randrange(1000)
+                print("a =", a, file=f)
+                print("b =", b, file=f)
 
             if TESTFN in sys.modules:
                 del sys.modules[TESTFN]