only close non-None files

This commit is contained in:
Benjamin Peterson 2013-05-11 16:29:03 -05:00
parent feb5149244
commit aa6f688bd4

View file

@ -241,7 +241,8 @@ def test_load_module_extension_file_is_None(self):
# Issue #15902 # Issue #15902
name = '_heapq' name = '_heapq'
found = imp.find_module(name) found = imp.find_module(name)
found[0].close() if found[0] is not None:
found[0].close()
if found[2][2] != imp.C_EXTENSION: if found[2][2] != imp.C_EXTENSION:
return return
imp.load_module(name, None, *found[1:]) imp.load_module(name, None, *found[1:])