[3.13] pythongh-123780: Make test_pkgutil clean up spam module (pythonGH-123036) (python#123781)

miss-islington · mhsmith · web-flow · commit 72cdd2ade6df · 2024-09-06T22:57:46.000+02:00
pythongh-123780: Make test_pkgutil clean up `spam` module (pythonGH-123036) (cherry picked from commit eca3fe4) Co-authored-by: Malcolm Smith <smith@chaquo.com>
diff --git a/Lib/test/test_pkgutil.py b/Lib/test/test_pkgutil.py
@@ -624,8 +624,11 @@ def test_get_loader_handles_missing_spec_attribute(self):
         mod = type(sys)(name)
         del mod.__spec__
         with CleanImport(name):
-            sys.modules[name] = mod
-            loader = pkgutil.get_loader(name)
+            try:
+                sys.modules[name] = mod
+                loader = pkgutil.get_loader(name)
+            finally:
+                sys.modules.pop(name, None)
         self.assertIsNone(loader)
 
     @ignore_warnings(category=DeprecationWarning)
@@ -634,8 +637,11 @@ def test_get_loader_handles_spec_attribute_none(self):
         mod = type(sys)(name)
         mod.__spec__ = None
         with CleanImport(name):
-            sys.modules[name] = mod
-            loader = pkgutil.get_loader(name)
+            try:
+                sys.modules[name] = mod
+                loader = pkgutil.get_loader(name)
+            finally:
+                sys.modules.pop(name, None)
         self.assertIsNone(loader)
 
     @ignore_warnings(category=DeprecationWarning)
