File tree 1 file changed +10
-4
lines changed
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):
624
624
mod = type (sys )(name )
625
625
del mod .__spec__
626
626
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 )
629
632
self .assertIsNone (loader )
630
633
631
634
@ignore_warnings (category = DeprecationWarning )
@@ -634,8 +637,11 @@ def test_get_loader_handles_spec_attribute_none(self):
634
637
mod = type (sys )(name )
635
638
mod .__spec__ = None
636
639
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 )
639
645
self .assertIsNone (loader )
640
646
641
647
@ignore_warnings (category = DeprecationWarning )
You can’t perform that action at this time.
0 commit comments