Module test_numba

Functions

▷ def test_serialize_numba(shape, dtype, order, serializers) 4 decorators
@pytest.mark.parametrize("shape", [(0,), (5,), (4, 6), (10, 11), (2, 3, 5)])
@pytest.mark.parametrize("dtype", ["u1", "u4", "u8", "f4"])
@pytest.mark.parametrize("order", ["C", "F"])
@pytest.mark.parametrize("serializers", [("cuda",), ("dask",)])
def test_serialize_numba(
shape,
dtype,
order,
serializers,
)
source link
▷ def test_serialize_numba_from_rmm(size) @pytest.mark.parametrize("size", [0, 3, 10])
@pytest.mark.parametrize("size", [0, 3, 10])
def test_serialize_numba_from_rmm(size)
source link