Fix a follow_imports (goto) issue.

2017-09-11 23:32:10 +02:00
parent 619acbd2ca
commit 4a544c29ea
2 changed files with 10 additions and 8 deletions
--- a/test/test_api/test_api.py
+++ b/test/test_api/test_api.py
@@ -207,6 +207,9 @@ def test_goto_assignments_follow_imports():
    definition, = script.goto_assignments()
    assert (definition.line, definition.column) == start_pos

+    d, = api.Script('a = 1\na').goto_assignments(follow_imports=True)
+    assert d.name == 'a'
+

 def test_goto_module():
    def check(line, expected):