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'