diff --git a/test/test_evaluate/test_imports.py b/test/test_evaluate/test_imports.py index 76a2499b..d0b65031 100644 --- a/test/test_evaluate/test_imports.py +++ b/test/test_evaluate/test_imports.py @@ -12,6 +12,7 @@ from jedi._compatibility import find_module_py33, find_module from jedi.evaluate import compiled from jedi.evaluate import imports from jedi.api.project import Project +from jedi.evaluate.gradual.stub_context import stub_to_actual_context_set from ..helpers import cwd_at, get_example_dir, test_dir, root_dir THIS_DIR = os.path.dirname(__file__) @@ -295,7 +296,10 @@ def test_compiled_import_none(monkeypatch, Script): """ script = Script('import sys') monkeypatch.setattr(compiled, 'load_module', lambda *args, **kwargs: None) - assert not script.goto_definitions() + def_, = script.goto_definitions() + assert def_.type == 'module' + context, = def_._name.infer() + assert not stub_to_actual_context_set(context) @pytest.mark.parametrize(