From f1465f0535ee64ee4bdcabe332c28943f579c672 Mon Sep 17 00:00:00 2001 From: Nick Coghlan Date: Mon, 15 Apr 2013 22:56:51 +1000 Subject: Close #17731: Clean up properly in test_import --- Lib/test/test_import.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/Lib/test/test_import.py b/Lib/test/test_import.py index 81ddf45..470a6d2 100644 --- a/Lib/test/test_import.py +++ b/Lib/test/test_import.py @@ -43,6 +43,7 @@ def _ready_to_import(name=None, source=""): # sets up a temporary directory and removes it # creates the module file # temporarily clears the module from sys.modules (if any) + # reverts or removes the module when cleaning up name = name or "spam" with script_helper.temp_dir() as tempdir: path = script_helper.make_script(tempdir, name, source) @@ -54,6 +55,8 @@ def _ready_to_import(name=None, source=""): finally: if old_module is not None: sys.modules[name] = old_module + elif name in sys.modules: + del sys.modules[name] class ImportTests(unittest.TestCase): -- cgit v0.12