conftest.py

import sys
import textwrap
from functools import partial
from packaging.version import Version
from platform import python_version

import pytest
import numpy as np

PYTEST_LT_8_5 = Version(pytest.__version__) < Version('8.5.0.dev')

# 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


# Windows + Python 3.14.0 + pytest-dev have ResourceWarning, see
# https://github.com/scientific-python/pytest-doctestplus/issues/305
def pytest_runtestloop(session):
    if sys.platform == 'win32' and python_version() == "3.14.0" and not PYTEST_LT_8_5:
        session.add_marker(pytest.mark.filterwarnings('ignore::ResourceWarning'))