| Index: third_party/logilab/logilab/astroid/brain/py2pytest.py
|
| diff --git a/third_party/logilab/logilab/astroid/brain/py2pytest.py b/third_party/logilab/logilab/astroid/brain/py2pytest.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e24d449cdc8254c586c7489888b84e650541102c
|
| --- /dev/null
|
| +++ b/third_party/logilab/logilab/astroid/brain/py2pytest.py
|
| @@ -0,0 +1,31 @@
|
| +"""Astroid hooks for pytest."""
|
| +
|
| +from astroid import MANAGER, register_module_extender
|
| +from astroid.builder import AstroidBuilder
|
| +
|
| +
|
| +def pytest_transform():
|
| + return AstroidBuilder(MANAGER).string_build('''
|
| +
|
| +try:
|
| + import _pytest.mark
|
| + import _pytest.recwarn
|
| + import _pytest.runner
|
| + import _pytest.python
|
| +except ImportError:
|
| + pass
|
| +else:
|
| + deprecated_call = _pytest.recwarn.deprecated_call
|
| + exit = _pytest.runner.exit
|
| + fail = _pytest.runner.fail
|
| + fixture = _pytest.python.fixture
|
| + importorskip = _pytest.runner.importorskip
|
| + mark = _pytest.mark.MarkGenerator()
|
| + raises = _pytest.python.raises
|
| + skip = _pytest.runner.skip
|
| + yield_fixture = _pytest.python.yield_fixture
|
| +
|
| +''')
|
| +
|
| +register_module_extender(MANAGER, 'pytest', pytest_transform)
|
| +register_module_extender(MANAGER, 'py.test', pytest_transform)
|
|
|