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

This commit is contained in:
Tim Peters 2001-03-04 00:30:25 +00:00
parent 0c8f6643c6
commit 243bff4708
1 changed files with 7 additions and 1 deletions

View File

@ -75,7 +75,9 @@ file = open(TESTFN, 'w')
file.write(source) file.write(source)
file.close() file.close()
# Note that load_source creates file TESTFN+'c' or TESTFN+'o'.
mod = imp.load_source('testmod', TESTFN) mod = imp.load_source('testmod', TESTFN)
files_to_clean_up = [TESTFN, TESTFN + 'c', TESTFN + 'o']
def istest(func, exp): def istest(func, exp):
obj = eval(exp) obj = eval(exp)
@ -204,4 +206,8 @@ test(inspect.formatargvalues(args, varargs, varkw, locals) ==
'(a=7, b=8, c=9, d=3, (e=4, (f=5,)), *g=(), **h={})', '(a=7, b=8, c=9, d=3, (e=4, (f=5,)), *g=(), **h={})',
'mod.fr.f_back formatted argvalues') 'mod.fr.f_back formatted argvalues')
os.unlink(TESTFN) for fname in files_to_clean_up:
try:
os.unlink(fname)
except:
pass