File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -1832,14 +1832,16 @@ def __init__(self, module_names):
1832
1832
self .module_names = module_names
1833
1833
self .fail_message = ''
1834
1834
if isinstance (module_names , str ):
1835
- self .module_names = set ([ module_names ])
1836
- sys . meta_path . insert ( 0 , self .ForbiddenModules (self , self .module_names ) )
1835
+ self .module_names = { module_names }
1836
+ self . meta_path_finder = self .ForbiddenModules (self , self .module_names )
1837
1837
1838
1838
def setUp (self ):
1839
1839
super ().setUp ()
1840
1840
self .addCleanup (self .cleanup )
1841
+ sys .meta_path .insert (0 , self .meta_path_finder )
1841
1842
1842
1843
def cleanup (self ):
1844
+ sys .meta_path .remove (self .meta_path_finder )
1843
1845
# We use a flag and check it during the cleanup phase to fail the test
1844
1846
# if needed. This is done because some module imports occur inside of a
1845
1847
# try-except block that ignores all exceptions, so raising an exception
You can’t perform that action at this time.
0 commit comments