forked from VimPlug/jedi
77 lines
3.0 KiB
Python
77 lines
3.0 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from jedi.api.project import Project
|
|
from test.helpers import root_dir
|
|
|
|
|
|
@pytest.mark.parametrize('type_', ['goto', 'infer'])
|
|
@pytest.mark.parametrize('way', ['direct', 'indirect'])
|
|
@pytest.mark.parametrize(
|
|
'kwargs', [
|
|
dict(only_stubs=False, prefer_stubs=False),
|
|
dict(only_stubs=False, prefer_stubs=True),
|
|
dict(only_stubs=True, prefer_stubs=False),
|
|
]
|
|
)
|
|
@pytest.mark.parametrize(
|
|
('code', 'full_name', 'has_stub', 'has_python', 'goto_changes'), [
|
|
['import os; os.walk', 'os.walk', True, True, {}],
|
|
['from collections import Counter', 'collections.Counter', True, True, {}],
|
|
['from collections', 'collections', True, True, {}],
|
|
['from collections import Counter; Counter', 'collections.Counter', True, True, {}],
|
|
['from collections import Counter; Counter()', 'collections.Counter', True, True, {}],
|
|
['from collections import Counter; Counter.most_common',
|
|
'collections.Counter.most_common', True, True, {}],
|
|
['from collections import deque', 'collections.deque', True, False, {}],
|
|
|
|
['from keyword import kwlist; kwlist', 'typing.Sequence', True, True,
|
|
{'full_name': 'keyword.kwlist'}],
|
|
['from keyword import kwlist', 'typing.Sequence', True, True,
|
|
{'full_name': 'keyword.kwlist'}],
|
|
|
|
['import with_stub', 'with_stub', True, True, {}],
|
|
['import with_stub', 'with_stub', True, True, {}],
|
|
['import with_stub_folder.python_only', 'with_stub_folder.python_only', False, True, {}],
|
|
['import stub_only', 'stub_only', True, False, {}],
|
|
])
|
|
def test_infer_and_goto(Script, code, full_name, has_stub, has_python, way,
|
|
kwargs, type_, goto_changes):
|
|
project = Project(os.path.join(root_dir, 'test', 'completion', 'stub_folder'))
|
|
s = Script(code, _project=project)
|
|
prefer_stubs = kwargs['prefer_stubs']
|
|
only_stubs = kwargs['only_stubs']
|
|
|
|
if type_ == 'goto':
|
|
full_name = goto_changes.get('full_name', full_name)
|
|
if way == 'direct':
|
|
if type_ == 'goto':
|
|
defs = s.goto_assignments(follow_imports=True, **kwargs)
|
|
else:
|
|
defs = s.goto_definitions(**kwargs)
|
|
else:
|
|
goto_defs = s.goto_assignments(
|
|
# Prefering stubs when we want to go to python and vice versa
|
|
prefer_stubs=not (prefer_stubs or only_stubs),
|
|
follow_imports=True,
|
|
)
|
|
if type_ == 'goto':
|
|
defs = [d for goto_def in goto_defs for d in goto_def.goto_assignments(**kwargs)]
|
|
else:
|
|
defs = [d for goto_def in goto_defs for d in goto_def.infer(**kwargs)]
|
|
|
|
if not has_stub and only_stubs:
|
|
assert not defs
|
|
else:
|
|
assert defs
|
|
|
|
for d in defs:
|
|
if prefer_stubs and has_stub:
|
|
assert d.is_stub()
|
|
if only_stubs:
|
|
assert d.is_stub()
|
|
assert d.full_name == full_name
|
|
#if d.is_stub():
|
|
#assert d.module_path.endswith('.pyi')
|