Fix the issues that were changed by removing start_pos from the api classes.

This commit is contained in:
Dave Halter
2017-01-04 22:24:25 +01:00
parent e96fd32588
commit 9341df11bf
4 changed files with 10 additions and 9 deletions

View File

@@ -211,7 +211,7 @@ class BaseDefinition(object):
start_pos = self._name.start_pos
if start_pos is None:
return None
return start_pos[0]
return start_pos[1]
def docstring(self, raw=False, fast=True):
r"""
@@ -558,8 +558,9 @@ class Definition(BaseDefinition):
:rtype: list of Definition
"""
defs = self._name.infer()
return _sort_names_by_start_pos(
common.unite(defined_names(self._evaluator, d) for d in defs)
return sorted(
common.unite(defined_names(self._evaluator, d) for d in defs),
key=lambda s: s._name.start_pos or (0, 0)
)
def is_definition(self):

View File

@@ -177,18 +177,18 @@ def test_goto_assignments_follow_imports():
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)
assert (definition.line, definition.column) == (1, 0)
definition, = api.Script(code).goto_assignments(follow_imports=True)
assert 'inspect.py' in definition.module_path
assert definition.start_pos > (1, 0)
assert (definition.line, definition.column) > (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.line, definition.column) == start_pos
assert definition.name == 'p'
result, = definition.goto_assignments()
assert result.name == 'p'
@@ -198,4 +198,4 @@ def test_goto_assignments_follow_imports():
assert result.name == 'int'
definition, = script.goto_assignments()
assert definition.start_pos == start_pos
assert (definition.line, definition.column) == start_pos

View File

@@ -272,7 +272,7 @@ class TestGotoAssignments(TestCase):
"""
bar = names(dedent(src), references=True)[-1]
param = bar.goto_assignments()[0]
assert param.start_pos == (1, 13)
assert (param.line, param.column) == (1, 13)
assert param.type == 'param'
def test_class_call(self):

View File

@@ -48,7 +48,7 @@ def test_nested_resolve():
cls = get_completion('XX', locals())
func = get_completion('XX.x', locals())
assert func.start_pos == (cls.start_pos[0] + 1, 12)
assert (func.line, func.column) == (cls.line + 1, 12)
def test_side_effect_completion():