Merge pull request #5743 from jdufresne/path-finder

Define _MockImporter as a MetaPathFinder
This commit is contained in:
Takeshi KOMIYA 2018-12-12 01:36:46 +09:00 committed by GitHub
commit aebfd0df71
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -98,7 +98,7 @@ class _MockModule(ModuleType):
return o
class _MockImporter:
class _MockImporter(MetaPathFinder):
def __init__(self, names):
# type: (List[str]) -> None
self.names = names
@ -120,7 +120,7 @@ class _MockImporter:
del sys.modules[m]
def find_module(self, name, path=None):
# type: (str, str) -> Any
# type: (str, Sequence[Union[bytes, str]]) -> Any
# check if name is (or is a descendant of) one of our base_packages
for n in self.names:
if n == name or name.startswith(n + '.'):