1
0
forked from VimPlug/jedi

Implement goto_assignments(follow_imports=True). Fixes #382.

This commit is contained in:
Dave Halter
2016-08-03 18:05:08 +02:00
parent c1bef454f5
commit ebd080a0fd
4 changed files with 37 additions and 7 deletions

View File

@@ -164,3 +164,16 @@ def test_get_line_code():
assert get_line_code(source, line=2) == line
assert get_line_code(source, line=2, after=1) == line + '\nother_line'
assert get_line_code(source, line=2, after=1, before=1) == source
def test_goto_assignments_follow_imports():
code = dedent("""
import inspect
inspect.isfunction""")
definition, = api.Script(code, column=0).goto_assignments(follow_imports=True)
assert 'inspect.py' in definition.module_path
assert definition.start_pos == (1, 0)
definition, = api.Script(code).goto_assignments(follow_imports=True)
assert 'inspect.py' in definition.module_path
assert definition.start_pos > (1, 0)