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
51
52
53
54
|
import textwrap
from functools import partial
from packaging.version import Version
import pytest
import numpy as np
# Keep this until we require numpy to be >=2.0 or there is a directive in doctestplus
# to support multiple ways of repr
if Version(np.__version__) >= Version("2.0.dev"):
np.set_printoptions(legacy="1.25")
def _wrap_docstring_in_func(func_name, docstring):
template = textwrap.dedent(r"""
def {}():
r'''
{}
'''
""")
return template.format(func_name, docstring)
@pytest.fixture
def makepyfile(testdir):
"""Fixture for making python files with single function and docstring."""
def make(*args, **kwargs):
func_name = kwargs.pop('func_name', 'f')
# content in args and kwargs is treated as docstring
wrap = partial(_wrap_docstring_in_func, func_name)
args = map(wrap, args)
kwargs = dict(zip(kwargs.keys(), map(wrap, kwargs.values())))
return testdir.makepyfile(*args, **kwargs)
return make
@pytest.fixture
def maketestfile(makepyfile):
"""Fixture for making python test files with single function and docstring."""
def make(*args, **kwargs):
func_name = kwargs.pop('func_name', 'test_foo')
return makepyfile(*args, func_name=func_name, **kwargs)
return make
@pytest.fixture
def makerstfile(testdir):
"""Fixture for making rst files with specified content."""
def make(*args, **kwargs):
return testdir.makefile('.rst', *args, **kwargs)
return make
|
