File tree Expand file tree Collapse file tree 1 file changed +10
-4
lines changed
Expand file tree Collapse file tree 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -624,8 +624,11 @@ def test_get_loader_handles_missing_spec_attribute(self):
624624 mod = type (sys )(name )
625625 del mod .__spec__
626626 with CleanImport (name ):
627- sys .modules [name ] = mod
628- loader = pkgutil .get_loader (name )
627+ try :
628+ sys .modules [name ] = mod
629+ loader = pkgutil .get_loader (name )
630+ finally :
631+ sys .modules .pop (name , None )
629632 self .assertIsNone (loader )
630633
631634 @ignore_warnings (category = DeprecationWarning )
@@ -634,8 +637,11 @@ def test_get_loader_handles_spec_attribute_none(self):
634637 mod = type (sys )(name )
635638 mod .__spec__ = None
636639 with CleanImport (name ):
637- sys .modules [name ] = mod
638- loader = pkgutil .get_loader (name )
640+ try :
641+ sys .modules [name ] = mod
642+ loader = pkgutil .get_loader (name )
643+ finally :
644+ sys .modules .pop (name , None )
639645 self .assertIsNone (loader )
640646
641647 @ignore_warnings (category = DeprecationWarning )
You can’t perform that action at this time.
0 commit comments