gvsig-scripting / org.gvsig.scripting / trunk / org.gvsig.scripting / org.gvsig.scripting.app / org.gvsig.scripting.app.mainplugin / src / main / resources-plugin / scripting / lib / astroid / brain / brain_pytest.py @ 745
History | View | Annotate | Download (908 Bytes)
| 1 |
"""Astroid hooks for pytest."""
|
|---|---|
| 2 |
from __future__ import absolute_import |
| 3 |
from astroid import MANAGER, register_module_extender |
| 4 |
from astroid.builder import AstroidBuilder |
| 5 |
|
| 6 |
|
| 7 |
def pytest_transform(): |
| 8 |
return AstroidBuilder(MANAGER).string_build(''' |
| 9 |
|
| 10 |
try:
|
| 11 |
import _pytest.mark
|
| 12 |
import _pytest.recwarn
|
| 13 |
import _pytest.runner
|
| 14 |
import _pytest.python
|
| 15 |
except ImportError:
|
| 16 |
pass
|
| 17 |
else:
|
| 18 |
deprecated_call = _pytest.recwarn.deprecated_call
|
| 19 |
exit = _pytest.runner.exit
|
| 20 |
fail = _pytest.runner.fail
|
| 21 |
fixture = _pytest.python.fixture
|
| 22 |
importorskip = _pytest.runner.importorskip
|
| 23 |
mark = _pytest.mark.MarkGenerator()
|
| 24 |
raises = _pytest.python.raises
|
| 25 |
skip = _pytest.runner.skip
|
| 26 |
yield_fixture = _pytest.python.yield_fixture
|
| 27 |
|
| 28 |
''')
|
| 29 |
|
| 30 |
register_module_extender(MANAGER, 'pytest', pytest_transform)
|
| 31 |
register_module_extender(MANAGER, 'py.test', pytest_transform)
|