added one more test

clEsperanto · Dec 2, 2022 · 0ccbeb3 · 0ccbeb3
1 parent 803e129
commit 0ccbeb3
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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],