VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-30 08:44:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Infer doctests and signatures uniformly, fixes #1466

Browse Source
This commit is contained in:
Dave Halter
2020-01-03 00:45:14 +01:00
parent 2d31e2e760
commit 3b6bbab556
10 changed files with 89 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
test/test_inference/test_gradual/test_stub_loading.py
View File

@@ -25,3 +25,41 @@ def ScriptInStubFolder(Script):
def test_find_stubs_infer(ScriptInStubFolder, code, expected):
defs = ScriptInStubFolder(code).infer()
assert [d.name for d in defs] == expected
func_without_stub_sig = 'func_without_stub(a)'
func_without_stub_doc = func_without_stub_sig + '\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', func_without_stub_sig),
('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', func_without_stub_sig),
('from with_python import func_without_stub as f; f', func_without_stub_sig),
('from with_python import func_without_stub; func_without_stub', func_without_stub_sig),
('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