def module_injector(inference_state, names, code):
assert isinstance(names, tuple)
file_io = KnownContentFileIO(
Path('foo/bar/module-injector-%s.py' % next(counter)).absolute(),
code)
v = _load_python_module(inference_state, file_io, names)
inference_state.module_cache.add(names, ValueSet([v]))
def test_get_modules_containing_name(inference_state, path, goal, is_package):
module = imports._load_python_module(
inference_state,
FileIO(path),
import_names=('ok', 'lala', 'x'),
is_package=is_package,
)
assert module
module_context = module.as_context()
input_module, found_module = imports.get_module_contexts_containing_name(
inference_state, [module_context], 'string_that_only_exists_here')
assert input_module is module_context
assert found_module.string_names == goal