ref: 3194ca04c3004ed8c840b775da9a3c0436feee32
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
def run_module_suite():
import sys, pytest
pytest.main(sys.argv)
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)
from numpy.testing import run_module_suite
# always use numpy's assert_equal
import numpy
assert_equal = numpy.testing.assert_equal