mirror of
https://github.com/python/cpython.git
synced 2025-09-28 03:13:48 +00:00
Merge for issue #14710
This commit is contained in:
commit
17f9cc07f3
2 changed files with 17 additions and 1 deletions
|
@ -456,6 +456,8 @@ def get_loader(module_or_name):
|
|||
"""
|
||||
if module_or_name in sys.modules:
|
||||
module_or_name = sys.modules[module_or_name]
|
||||
if module_or_name is None:
|
||||
return None
|
||||
if isinstance(module_or_name, ModuleType):
|
||||
module = module_or_name
|
||||
loader = getattr(module, '__loader__', None)
|
||||
|
@ -487,7 +489,7 @@ def find_loader(fullname):
|
|||
# pkgutil previously raised ImportError
|
||||
msg = "Error while finding loader for {!r} ({}: {})"
|
||||
raise ImportError(msg.format(fullname, type(ex), ex)) from ex
|
||||
return spec.loader
|
||||
return spec.loader if spec is not None else None
|
||||
|
||||
|
||||
def extend_path(path, name):
|
||||
|
|
|
@ -363,6 +363,20 @@ class ImportlibMigrationTests(unittest.TestCase):
|
|||
loader = pkgutil.get_loader(name)
|
||||
self.assertIsNone(loader)
|
||||
|
||||
def test_get_loader_None_in_sys_modules(self):
|
||||
name = 'totally bogus'
|
||||
sys.modules[name] = None
|
||||
try:
|
||||
loader = pkgutil.get_loader(name)
|
||||
finally:
|
||||
del sys.modules[name]
|
||||
self.assertIsNone(loader)
|
||||
|
||||
def test_find_loader_missing_module(self):
|
||||
name = 'totally bogus'
|
||||
loader = pkgutil.find_loader(name)
|
||||
self.assertIsNone(loader)
|
||||
|
||||
def test_find_loader_avoids_emulation(self):
|
||||
with check_warnings() as w:
|
||||
self.assertIsNotNone(pkgutil.find_loader("sys"))
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue