def test_copy_slice_to_3d():
test1 = cle.push(np.asarray([[3, 4], [4, 5]]))
test2 = cle.create((2, 2, 2))
cle.set(test2, 0)
cle.copy_slice(test1, test2, 0)
print(test2)
a = cle.pull(test2)
assert (np.min(a) == 0)
assert (np.max(a) == 5)
assert (np.mean(a) == 2)
def test_copy_slice_from_3d():
test1 = cle.push(np.asarray([[[1, 4], [0, 4]], [[1, 3], [1, 2]]]))
test2 = cle.create((2, 2))
cle.copy_slice(test1, test2, 0)
print(test2)
a = cle.pull(test2)
assert (np.min(a) == 0)
assert (np.max(a) == 1)
assert (np.mean(a) == 0.75)
def test_copy_slice_mini_x():
np_input = np.asarray([[1, 2, 3, 4]])
gpu_input = cle.push_zyx(np_input)
gpu_output = cle.create((1, 1, 4))
cle.copy_slice(gpu_input, gpu_output, 0)
print(gpu_output)
a = cle.pull(gpu_output)
assert (np.min(a) == 1)
assert (np.max(a) == 4)
assert (np.mean(a) == 2.5)
def test_copy_slice_mini_y():
np_input = np.asarray([[1], [2], [3], [4]])
gpu_input = cle.push(np_input)
gpu_output = cle.create((1, 4, 1))
cle.copy_slice(gpu_input, gpu_output, 0)
print(gpu_output)
a = cle.pull(gpu_output)
assert (np.min(a) == 1)
assert (np.max(a) == 4)
assert (np.mean(a) == 2.5)
def test_copy_slice_to3d_with_one_slice_zyx():
test1 = cle.push_zyx(np.asarray([[3, 4, 6], [4, 5, 2]]))
print(test1)
print("shape test1 " + str(test1.shape))
test2 = cle.create((1, 2, 3))
print("shape test2 " + str(test2.shape))
print(test2)
cle.copy_slice(test1, test2, 0)
print(test2)
a = cle.pull(test2)
assert (np.min(a) == 2)
assert (np.max(a) == 6)
assert (np.mean(a) == 4)