source: python/tests/_tools.py @ cdfe9ce

"""
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