mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 14:04:26 +08:00
4b276bae87
With this change we can now include all parents of the script, which will make relative imports always work. Now the whole meta_path is scanned and not just importlib's PathFinder. Fixes #1183.
94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
from os.path import dirname, join
|
|
|
|
import pytest
|
|
import py
|
|
|
|
from ..helpers import get_example_dir
|
|
|
|
|
|
SYS_PATH = [join(dirname(__file__), d)
|
|
for d in ['namespace_package/ns1', 'namespace_package/ns2']]
|
|
|
|
|
|
def script_with_path(Script, *args, **kwargs):
|
|
return Script(sys_path=SYS_PATH, *args, **kwargs)
|
|
|
|
|
|
def test_goto_definition(Script):
|
|
assert script_with_path(Script, 'from pkg import ns1_file').goto_definitions()
|
|
assert script_with_path(Script, 'from pkg import ns2_file').goto_definitions()
|
|
assert not script_with_path(Script, 'from pkg import ns3_file').goto_definitions()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
('source', 'solution'), [
|
|
('from pkg.ns2_folder.nested import foo', 'nested!'),
|
|
('from pkg.ns2_folder import foo', 'ns2_folder!'),
|
|
('from pkg.ns2_file import foo', 'ns2_file!'),
|
|
('from pkg.ns1_folder import foo', 'ns1_folder!'),
|
|
('from pkg.ns1_file import foo', 'ns1_file!'),
|
|
('from pkg import foo', 'ns1!'),
|
|
]
|
|
)
|
|
def test_goto_assignment(Script, source, solution):
|
|
ass = script_with_path(Script, source).goto_assignments()
|
|
assert len(ass) == 1
|
|
assert ass[0].description == "foo = '%s'" % solution
|
|
|
|
|
|
def test_simple_completions(Script):
|
|
# completion
|
|
completions = script_with_path(Script, 'from pkg import ').completions()
|
|
names = [str(c.name) for c in completions] # str because of unicode
|
|
compare = ['foo', 'ns1_file', 'ns1_folder', 'ns2_folder', 'ns2_file',
|
|
'pkg_resources', 'pkgutil', '__name__', '__path__',
|
|
'__package__', '__file__', '__doc__']
|
|
# must at least contain these items, other items are not important
|
|
assert set(compare) == set(names)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
('source', 'solution'), [
|
|
('from pkg import ns2_folder as x', 'ns2_folder!'),
|
|
('from pkg import ns2_file as x', 'ns2_file!'),
|
|
('from pkg.ns2_folder import nested as x', 'nested!'),
|
|
('from pkg import ns1_folder as x', 'ns1_folder!'),
|
|
('from pkg import ns1_file as x', 'ns1_file!'),
|
|
('import pkg as x', 'ns1!'),
|
|
]
|
|
)
|
|
def test_completions(Script, source, solution):
|
|
for c in script_with_path(Script, source + '; x.').completions():
|
|
if c.name == 'foo':
|
|
completion = c
|
|
solution = "foo = '%s'" % solution
|
|
assert completion.description == solution
|
|
|
|
|
|
def test_nested_namespace_package(Script):
|
|
code = 'from nested_namespaces.namespace.pkg import CONST'
|
|
|
|
sys_path = [dirname(__file__)]
|
|
|
|
script = Script(sys_path=sys_path, source=code, line=1, column=45)
|
|
|
|
result = script.goto_definitions()
|
|
|
|
assert len(result) == 1
|
|
|
|
|
|
def test_relative_import(Script, tmpdir):
|
|
"""
|
|
Attempt a relative import in a very simple namespace package.
|
|
"""
|
|
directory = get_example_dir('namespace_package_relative_import')
|
|
# Need to copy the content in a directory where there's no __init__.py.
|
|
py.path.local(directory).copy(tmpdir)
|
|
file_path = join(tmpdir.strpath, "rel1.py")
|
|
script = Script(path=file_path, line=1)
|
|
d, = script.goto_definitions()
|
|
assert d.name == 'int'
|
|
d, = script.goto_assignments()
|
|
assert d.name == 'name'
|
|
assert d.module_name == 'rel2'
|