From 0ccbeb35e6cf4eda5c7b5ef574eed79b4fd3f465 Mon Sep 17 00:00:00 2001 From: Robert Haase Date: Fri, 2 Dec 2022 08:15:40 +0100 Subject: [PATCH] added one more test --- tests/test_maximum_position.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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],