diff options
Diffstat (limited to 'Lib/test/test_import.py')
| -rw-r--r-- | Lib/test/test_import.py | 3 |
1 files changed, 3 insertions, 0 deletions
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): |