VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

Fix alias usages in goto_assignments.

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
jedi/evaluate/__init__.py
View File

@@ -324,6 +324,8 @@ class Evaluator(object):
return [call_path[0]] return [call_path[0]]
names = stmt.get_all_import_names() names = stmt.get_all_import_names()
if stmt.alias_name_part:
names = names[:-1]
# Filter names that are after our Name # Filter names that are after our Name
removed_names = len(names) - names.index(call_path[0]) - 1 removed_names = len(names) - names.index(call_path[0]) - 1
i = imports.ImportWrapper(self, stmt, kill_count=removed_names, i = imports.ImportWrapper(self, stmt, kill_count=removed_names,

12
test/test_api/test_api_classes.py
View File

@@ -286,3 +286,15 @@ class TestGotoAssignments(TestCase):
n = nms[1].goto_assignments()[0] n = nms[1].goto_assignments()[0]
assert n.name == 'path' assert n.name == 'path'
assert n.type == 'import' 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()