def test_with_scipy():
with use_kernel('scipy'):
k = get_kernel()
assert k.__name__ == 'csr.kernels.scipy'
k = get_kernel()
assert k is default_kernel
def test_get_mkl():
try:
import csr.kernels.mkl # noqa: F401
except ImportError:
pytest.skip("kernel MKL is disabled")
k = get_kernel('mkl')
assert k.__name__ == 'csr.kernels.mkl'
def test_get_default():
k = get_kernel()
assert k is not None
assert k is default_kernel
# make sure the default is right
try:
import csr.kernels.mkl # noqa: F401
assert k.__name__ == 'csr.kernels.mkl'
except ImportError:
assert k.__name__ == 'csr.kernels.numba'
def kernel(request):
"""
Fixture for variable CSR kernels. This fixture is parameterized, so if you
write a test function with a parameter ``kernel`` as its first parameter, it
will be called once for each kernel under active test.
"""
if request.param in DISABLED_KERNELS:
skip(f'kernel {request.param} is disabled')
with use_kernel(request.param):
k = get_kernel()
# warm-up the kernel
m = CSR.empty(1, 1)
h = k.to_handle(m)
k.release_handle(h)
del h, m
yield k
def test_get_numba():
k = get_kernel('numba')
assert k.__name__ == 'csr.kernels.numba'
def test_get_scipy():
k = get_kernel('scipy')
assert k.__name__ == 'csr.kernels.scipy'