Close #17731: Clean up properly in test_import
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 @@
# 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 @@
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):