diff options
Diffstat (limited to 'Lib/test/test_modulefinder.py')
-rw-r--r-- | Lib/test/test_modulefinder.py | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/Lib/test/test_modulefinder.py b/Lib/test/test_modulefinder.py index b4fa06a..a184217 100644 --- a/Lib/test/test_modulefinder.py +++ b/Lib/test/test_modulefinder.py @@ -204,11 +204,17 @@ def open_file(path): def create_package(source): ofi = None - for line in source.splitlines(): - if line.startswith(" ") or line.startswith("\t"): - ofi.write(line.strip() + "\n") - else: - ofi = open_file(os.path.join(TEST_DIR, line.strip())) + try: + for line in source.splitlines(): + if line.startswith(" ") or line.startswith("\t"): + ofi.write(line.strip() + "\n") + else: + if ofi: + ofi.close() + ofi = open_file(os.path.join(TEST_DIR, line.strip())) + finally: + if ofi: + ofi.close() class ModuleFinderTest(unittest.TestCase): def _do_test(self, info, report=False): |