Fix param issues in goto definition.

This commit is contained in:
Dave Halter
2017-01-04 08:58:29 +01:00
parent 24457bfe2e
commit cd23499fbe
4 changed files with 29 additions and 5 deletions

View File

@@ -177,3 +177,20 @@ def test_goto_assignments_follow_imports():
definition, = api.Script(code).goto_assignments(follow_imports=True)
assert 'inspect.py' in definition.module_path
assert definition.start_pos > (1, 0)
code = '''def param(p): pass\nparam(1)'''
start_pos = 1, len('def param(')
script = api.Script(code, *start_pos)
definition, = script.goto_assignments(follow_imports=True)
assert definition.start_pos == start_pos
assert definition.name == 'p'
result, = definition.goto_assignments()
assert result.name == 'p'
result, = definition._goto_definitions()
assert result.name == 'int'
result, = result._goto_definitions()
assert result.name == 'int'
definition, = script.goto_assignments()
assert definition.start_pos == start_pos