Clean up junk files left behind by imp.load_source().

tim-one · tim-one · commit 243bff470854 · 2001-03-04T00:30:25.000Z
diff --git a/Lib/test/test_inspect.py b/Lib/test/test_inspect.py
@@ -75,7 +75,9 @@ def test(assertion, message, *args):
 file.write(source)
 file.close()
 
+# Note that load_source creates file TESTFN+'c' or TESTFN+'o'.
 mod = imp.load_source('testmod', TESTFN)
+files_to_clean_up = [TESTFN, TESTFN + 'c', TESTFN + 'o']
 
 def istest(func, exp):
     obj = eval(exp)
@@ -204,4 +206,8 @@ def sourcerange(top, bottom):
      '(a=7, b=8, c=9, d=3, (e=4, (f=5,)), *g=(), **h={})',
      'mod.fr.f_back formatted argvalues')
 
-os.unlink(TESTFN)
+for fname in files_to_clean_up:
+    try:
+        os.unlink(fname)
+    except:
+        pass
