diff options
| author | Nick Coghlan <ncoghlan@gmail.com> | 2013-04-15 12:56:51 (GMT) |
|---|---|---|
| committer | Nick Coghlan <ncoghlan@gmail.com> | 2013-04-15 12:56:51 (GMT) |
| commit | f1465f0535ee64ee4bdcabe332c28943f579c672 (patch) | |
| tree | 8146b433aa25ccb2d34469aca450d6d106952581 | |
| parent | 26dfaac9ac0cf9a34006ab52a0e0cc4e39b772fa (diff) | |
| download | cpython-f1465f0535ee64ee4bdcabe332c28943f579c672.zip cpython-f1465f0535ee64ee4bdcabe332c28943f579c672.tar.gz cpython-f1465f0535ee64ee4bdcabe332c28943f579c672.tar.bz2 | |
Close #17731: Clean up properly in test_import
| -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): |