Function test_cupy_containers

dali/test/python/test_external_source_impl_utils.py:100–106

()

@attr("cupy")
def test_cupy_containers():
    import cupy as cp

    test_array = cp.array([[42, 42], [42, 42]], dtype=cp.uint8)
    disallowed_samples = [test_array, tensors.TensorGPU(test_array)]
    disallowed_batches = [tensors.TensorListGPU(test_array)]
    yield from run_checks([], [], disallowed_samples, disallowed_batches)