diff --git a/jedi/inference/names.py b/jedi/inference/names.py index 7e4da7db..81be3f96 100644 --- a/jedi/inference/names.py +++ b/jedi/inference/names.py @@ -133,7 +133,7 @@ class AbstractTreeName(AbstractNameDefinition): def get_defining_qualified_value(self): if self.is_import(): - raise 1 + raise NotImplementedError("Shouldn't really happen, please report") elif self.parent_context: return self.parent_context.get_value() # Might be None return None diff --git a/test/test_inference/test_gradual/test_conversion.py b/test/test_inference/test_gradual/test_conversion.py index 1e05e5cc..a86c8351 100644 --- a/test/test_inference/test_gradual/test_conversion.py +++ b/test/test_inference/test_gradual/test_conversion.py @@ -54,3 +54,11 @@ def test_goto_on_file(Script): script = Script(path=d1.module_path, _project=project) d2, = script.goto(line=d1.line, column=d1.column + 4) assert d2.name == 'Bar' + + +def test_goto_import(Script): + code = 'from abc import ABC; ABC' + d, = Script(code).goto(only_stubs=True) + assert d.is_stub() + d, = Script(code).goto() + assert not d.is_stub()