Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-11-04 07:31:38 +00:00
Code Issues Projects Releases Packages Wiki Activity

Change import_fresh_module to work with packages.

Browse source
This commit is contained in:
Ezio Melotti 2011-05-14 06:02:25 +03:00
parent 7c40489180
commit fec3ad1036
1 changed files with 6 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

14
Lib/test/support.py
View file

@ -80,19 +80,15 @@ def import_module(name, deprecated=False):
def _save_and_remove_module(name, orig_modules):
"""Helper function to save and remove a module from sys.modules
Return True if the module was in sys.modules, False otherwise.
Raise ImportError if the module can't be imported."""
saved = True
try:
orig_modules[name] = sys.modules[name]
except KeyError:
# try to import the module and raise an error if it can't be imported
if name not in sys.modules:
__import__(name)
saved = False
else:
del sys.modules[name]
return saved
for modname in list(sys.modules):
if modname == name or modname.startswith(name + '.'):
orig_modules[modname] = sys.modules[modname]
del sys.modules[modname]
def _save_and_block_module(name, orig_modules):
"""Helper function to save and block a module in sys.modules