mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 14:04:26 +08:00
65 lines
2.8 KiB
Python
65 lines
2.8 KiB
Python
from functools import partial
|
|
from test.helpers import get_example_dir
|
|
from jedi.api.project import Project
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def ScriptInStubFolder(Script):
|
|
path = get_example_dir('stub_packages')
|
|
project = Project(path, sys_path=[path], smart_sys_path=False)
|
|
return partial(Script, project=project)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
('code', 'expected'), [
|
|
('from no_python import foo', ['int']),
|
|
('from with_python import stub_only', ['str']),
|
|
('from with_python import python_only', ['int']),
|
|
('from with_python import both', ['int']),
|
|
('from with_python import something_random', []),
|
|
('from with_python.module import in_sub_module', ['int']),
|
|
]
|
|
)
|
|
def test_find_stubs_infer(ScriptInStubFolder, code, expected):
|
|
defs = ScriptInStubFolder(code).infer()
|
|
assert [d.name for d in defs] == expected
|
|
|
|
|
|
func_without_stub_doc = 'func_without_stub(a)\n\nnostubdoc'
|
|
func_with_stub_doc = 'func_with_stub(b: int) -> float\n\nwithstubdoc'
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
('code', 'expected'), [
|
|
('from with_python import stub_only', ''),
|
|
('from with_python import python_only', ''),
|
|
('from with_python import both', ''),
|
|
|
|
('import with_python; with_python.func_without_stub', ''),
|
|
('import with_python.module; with_python.module.func_without_stub', func_without_stub_doc),
|
|
('from with_python import module; module.func_without_stub', func_without_stub_doc),
|
|
('from with_python.module import func_without_stub', func_without_stub_doc),
|
|
('from with_python.module import func_without_stub as f; f', func_without_stub_doc),
|
|
('from with_python.module import func_without_stub; func_without_stub',
|
|
func_without_stub_doc),
|
|
('from with_python import func_without_stub', ''),
|
|
('from with_python import func_without_stub as f; f', ''),
|
|
('from with_python import func_without_stub; func_without_stub', ''),
|
|
|
|
('import with_python; with_python.func_with_stub', func_with_stub_doc),
|
|
('import with_python.module; with_python.module.func_with_stub', func_with_stub_doc),
|
|
('from with_python import module; module.func_with_stub', func_with_stub_doc),
|
|
('from with_python.module import func_with_stub', func_with_stub_doc),
|
|
('from with_python.module import func_with_stub as f; f', func_with_stub_doc),
|
|
('from with_python.module import func_with_stub; func_with_stub', func_with_stub_doc),
|
|
('from with_python import func_with_stub', func_with_stub_doc),
|
|
('from with_python import func_with_stub as f; f', func_with_stub_doc),
|
|
('from with_python import func_with_stub; func_with_stub', func_with_stub_doc),
|
|
]
|
|
)
|
|
def test_docstrings(ScriptInStubFolder, code, expected):
|
|
d, = ScriptInStubFolder(code).help()
|
|
assert d.docstring() == expected
|