def test_init_with_device(shape, dtype_spec, device): array = chainerx.ndarray(shape, dtype_spec, device=device) _check_array(array, dtype_spec, shape, device=device)
def test_init(shape, dtype_spec): array = chainerx.ndarray(shape, dtype_spec) _check_array(array, dtype_spec, shape)