diff options
| author | Malcolm Smith <smith@chaquo.com> | 2024-09-06 13:23:55 (GMT) |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-06 13:23:55 (GMT) |
| commit | eca3fe40c251d51964172dd4e6e9c7d0d85d7d4a (patch) | |
| tree | 41a3e1848a6b2f037c3bd288d98ad9370a6470a5 /Lib | |
| parent | 782a076362ca1836c2052652b1a3a352dec45186 (diff) | |
| download | cpython-eca3fe40c251d51964172dd4e6e9c7d0d85d7d4a.zip cpython-eca3fe40c251d51964172dd4e6e9c7d0d85d7d4a.tar.gz cpython-eca3fe40c251d51964172dd4e6e9c7d0d85d7d4a.tar.bz2 | |
gh-123780: Make test_pkgutil clean up `spam` module (GH-123036)
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/test/test_pkgutil.py | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/Lib/test/test_pkgutil.py b/Lib/test/test_pkgutil.py index 20fba87..ca69275 100644 --- a/Lib/test/test_pkgutil.py +++ b/Lib/test/test_pkgutil.py @@ -624,8 +624,11 @@ class ImportlibMigrationTests(unittest.TestCase): 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 @@ class ImportlibMigrationTests(unittest.TestCase): 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) |