ref: 00fcc5d123fafbb9c424d7725fd1ded52f25de5e
dir: /python/tests/_tools.py/
"""
This file imports test methods from different testing modules, in this
order:
- try importing 'pytest'
- if it fails, fallback to 'numpy.testing'
Nose2 support was removed because of lacking assertWarns on py2.7.
"""
import sys
_has_pytest = False
# check if we have pytest
try:
import pytest
parametrize = pytest.mark.parametrize
assert_raises = pytest.raises
assert_warns = pytest.warns
skipTest = pytest.skip
_has_pytest = True
except:
pass
# otherwise fallback on numpy.testing
if not _has_pytest:
from numpy.testing import dec, assert_raises, assert_warns
from numpy.testing import SkipTest
parametrize = dec.parametrize
def skipTest(msg):
raise SkipTest(msg)
# always use numpy's assert_equal
import numpy
assert_equal = numpy.testing.assert_equal