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

Add support for pytest fixtures from local pytest plugins.

Browse Source
This commit is contained in:
Martin Vielsmaier
2023-01-31 21:41:39 +01:00
parent 11280ef502
commit 7d77f61040
4 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
jedi/plugins/pytest.py
View File

@@ -181,6 +181,9 @@ def _iter_pytest_modules(module_context, skip_own_module=False):
if Path(file_io.path) != module_context.py__file__():
try:
m = load_module_from_path(module_context.inference_state, file_io)
pytest_plugins_pointer = m.goto("pytest_plugins")
if pytest_plugins_pointer:
yield from _load_pytest_plugins(module_context, pytest_plugins_pointer[0])
yield m.as_context()
except FileNotFoundError:
pass
@@ -196,6 +199,17 @@ def _iter_pytest_modules(module_context, skip_own_module=False):
yield module_value.as_context()
def _load_pytest_plugins(module_context, plugins_pointers):
from jedi.inference.value import iterable
from parso.python.tree import String
for inferred in plugins_pointers.infer():
if isinstance(inferred, iterable.SequenceLiteralValue):
for value in inferred.get_tree_entries():
if isinstance(value, String):
for module_value in module_context.inference_state.import_module(eval(value.value).split(".")):
yield module_value.as_context()
class FixtureFilter(ParserTreeFilter):
def _filter(self, names):
for name in super()._filter(names):