blob: b5537c099a569616e9d7e154db9d388408bc9dca (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
"""Astroid hooks for pytest."""
from astroid import MANAGER
from astroid import nodes
from astroid.builder import AstroidBuilder
MODULE_TRANSFORMS = {}
def transform(module):
try:
tr = MODULE_TRANSFORMS[module.name]
except KeyError:
pass
else:
tr(module)
def pytest_transform(module):
fake = 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
''')
for item_name, item in fake.locals.items():
module.locals[item_name] = item
MODULE_TRANSFORMS['pytest'] = pytest_transform
MODULE_TRANSFORMS['py.test'] = pytest_transform
MANAGER.register_transform(nodes.Module, transform)
|
