Skip to content

Commit 89c64b5

Browse files
Adjust initialization script for removal of numpy._testing.nosetests
1 parent 5b2ee1a commit 89c64b5

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

mkl_random/__init__.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,14 @@
2929
from .mklrand import *
3030
from ._version import __version__
3131

32-
from numpy.testing.nosetester import _numpy_tester
33-
test = _numpy_tester().test
34-
bench = _numpy_tester().bench
32+
try:
33+
from numpy.testing.nosetester import _numpy_tester
34+
test = _numpy_tester().test
35+
bench = _numpy_tester().bench
36+
del _numpy_tester
37+
except ModuleNotFoundError:
38+
# Pytest testing
39+
from numpy._pytesttester import PytestTester
40+
test = PytestTester(__name__)
41+
del PytestTester
42+

0 commit comments

Comments
 (0)