diff --git a/tests/test_maximum_position.py b/tests/test_maximum_position.py index 93cff9cc..886c6766 100644 --- a/tests/test_maximum_position.py +++ b/tests/test_maximum_position.py @@ -22,6 +22,18 @@ def test_maximum_position_3d(): assert (result == reference) +def test_maximum_position_3d_2(): + np_input = np.zeros((10,11,12)) + + np_input[1, 2, 3] = 1 + np_input[6, 7, 8] = 1 + + reference = ndimage.maximum_position(np_input) + gpu_input = cle.push(np_input) + result = cle.maximum_position(gpu_input) + + assert (result == reference) + def test_maximum_position_2d(): np_input= np.asarray([ [1, 2, 3, 10],