forked from VimPlug/jedi
Fix alias usages in goto_assignments.
This commit is contained in:
@@ -324,6 +324,8 @@ class Evaluator(object):
|
||||
return [call_path[0]]
|
||||
|
||||
names = stmt.get_all_import_names()
|
||||
if stmt.alias_name_part:
|
||||
names = names[:-1]
|
||||
# Filter names that are after our Name
|
||||
removed_names = len(names) - names.index(call_path[0]) - 1
|
||||
i = imports.ImportWrapper(self, stmt, kill_count=removed_names,
|
||||
|
||||
@@ -286,3 +286,15 @@ class TestGotoAssignments(TestCase):
|
||||
n = nms[1].goto_assignments()[0]
|
||||
assert n.name == 'path'
|
||||
assert n.type == 'import'
|
||||
|
||||
def test_import_alias(self):
|
||||
nms = names('import json as foo', references=True)
|
||||
assert nms[0].name == 'json'
|
||||
assert nms[0].type == 'import'
|
||||
n = nms[0].goto_assignments()[0]
|
||||
assert n.name == 'json'
|
||||
assert n.type == 'module'
|
||||
|
||||
assert nms[1].name == 'foo'
|
||||
assert nms[1].type == 'import'
|
||||
assert [nms[1]] == nms[1].goto_assignments()
|
||||
|
||||
Reference in New Issue
Block a user