VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

Fix completion for non-pytest params

Browse Source
This commit is contained in:
Dave Halter
2019-12-27 13:02:16 +01:00
parent 31936776a5
commit 4c22f4dbb1
3 changed files with 59 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
jedi/plugins/pytest.py
View File

@@ -54,14 +54,16 @@ def goto_anonymous_param(func):
def complete_param_names(func):
def wrapper(context, func_name):
def wrapper(context, func_name, decorator_nodes):
module_context = context.get_root_context()
names = []
for module_context in _iter_pytest_modules(module_context):
names += FixtureFilter(module_context).values()
if names:
return names
return func(context, func_name)
if func_name.startswith('test_') \
or any('fixture' in n.get_code() for n in decorator_nodes):
names = []
for module_context in _iter_pytest_modules(module_context):
names += FixtureFilter(module_context).values()
if names:
return names
return func(context, func_name, decorator_nodes)
return wrapper