diff options
Diffstat (limited to 'numpy/conftest.py')
-rw-r--r-- | numpy/conftest.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/numpy/conftest.py b/numpy/conftest.py index d3db866a7..15985a75b 100644 --- a/numpy/conftest.py +++ b/numpy/conftest.py @@ -6,6 +6,7 @@ from __future__ import division, absolute_import, print_function import warnings import pytest import numpy +import importlib from numpy.core.multiarray_tests import get_fpu_mode @@ -73,3 +74,13 @@ def pytest_collection_modifyitems(config, items): @pytest.fixture(autouse=True) def add_np(doctest_namespace): doctest_namespace['np'] = numpy + + +for module, replacement in { + 'numpy.testing.decorators': 'numpy.testing.pytest_tools.decorators', + 'numpy.testing.utils': 'numpy.testing.pytest_tools.utils', +}.items(): + module = importlib.import_module(module) + replacement = importlib.import_module(replacement) + module.__dict__.clear() + module.__dict__.update(replacement.__dict__) |