1
0
forked from VimPlug/jedi

Fix alias usages in goto_assignments.

This commit is contained in:
Dave Halter
2014-09-19 13:42:47 +02:00
parent ed56f73836
commit 99d35e57b6
2 changed files with 14 additions and 0 deletions

View File

@@ -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,

View File

@@ -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()