85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
|
# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
|
||
|
|
# For details: https://github.com/pylint-dev/astroid/blob/main/LICENSE
|
||
|
|
# Copyright (c) https://github.com/pylint-dev/astroid/blob/main/CONTRIBUTORS.txt
|
||
|
|
|
||
|
|
"""Astroid hooks for pytest."""
|
||
|
|
from astroid.brain.helpers import register_module_extender
|
||
|
|
from astroid.builder import AstroidBuilder
|
||
|
|
from astroid.manager import AstroidManager
|
||
|
|
|
||
|
|
|
||
|
|
def pytest_transform():
|
||
|
|
return AstroidBuilder(AstroidManager()).string_build(
|
||
|
|
"""
|
||
|
|
|
||
|
|
try:
|
||
|
|
import _pytest.mark
|
||
|
|
import _pytest.recwarn
|
||
|
|
import _pytest.runner
|
||
|
|
import _pytest.python
|
||
|
|
import _pytest.skipping
|
||
|
|
import _pytest.assertion
|
||
|
|
except ImportError:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
deprecated_call = _pytest.recwarn.deprecated_call
|
||
|
|
warns = _pytest.recwarn.warns
|
||
|
|
|
||
|
|
exit = _pytest.runner.exit
|
||
|
|
fail = _pytest.runner.fail
|
||
|
|
skip = _pytest.runner.skip
|
||
|
|
importorskip = _pytest.runner.importorskip
|
||
|
|
|
||
|
|
xfail = _pytest.skipping.xfail
|
||
|
|
mark = _pytest.mark.MarkGenerator()
|
||
|
|
raises = _pytest.python.raises
|
||
|
|
|
||
|
|
# New in pytest 3.0
|
||
|
|
try:
|
||
|
|
approx = _pytest.python.approx
|
||
|
|
register_assert_rewrite = _pytest.assertion.register_assert_rewrite
|
||
|
|
except AttributeError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
# Moved in pytest 3.0
|
||
|
|
|
||
|
|
try:
|
||
|
|
import _pytest.freeze_support
|
||
|
|
freeze_includes = _pytest.freeze_support.freeze_includes
|
||
|
|
except ImportError:
|
||
|
|
try:
|
||
|
|
import _pytest.genscript
|
||
|
|
freeze_includes = _pytest.genscript.freeze_includes
|
||
|
|
except ImportError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
try:
|
||
|
|
import _pytest.debugging
|
||
|
|
set_trace = _pytest.debugging.pytestPDB().set_trace
|
||
|
|
except ImportError:
|
||
|
|
try:
|
||
|
|
import _pytest.pdb
|
||
|
|
set_trace = _pytest.pdb.pytestPDB().set_trace
|
||
|
|
except ImportError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
try:
|
||
|
|
import _pytest.fixtures
|
||
|
|
fixture = _pytest.fixtures.fixture
|
||
|
|
yield_fixture = _pytest.fixtures.yield_fixture
|
||
|
|
except ImportError:
|
||
|
|
try:
|
||
|
|
import _pytest.python
|
||
|
|
fixture = _pytest.python.fixture
|
||
|
|
yield_fixture = _pytest.python.yield_fixture
|
||
|
|
except ImportError:
|
||
|
|
pass
|
||
|
|
"""
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def register(manager: AstroidManager) -> None:
|
||
|
|
register_module_extender(manager, "pytest", pytest_transform)
|
||
|
|
register_module_extender(manager, "py.test", pytest_transform)
|