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