fix: adapt seed pixel determination test according to changes

Kohulan · Oct 19, 2023 · 337775d · 337775d
1 parent f1deb46
commit 337775d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_mask_expansion.py b/tests/test_mask_expansion.py
@@ -117,10 +117,11 @@ def test_get_mask_center():
 
 
 def test_get_seeds():
-    test_image_array = np.array([(3, 2)])
-    test_mask_array = np.array([(9, 5, 9)])
+    test_image_array = np.array([[0, 1, 0],[1, 0, 1],[0, 1, 0]])
+    test_mask_array = np.ones(test_image_array.shape)
+    exclusion_mask = np.zeros(test_image_array.shape)
     expected_result = []
-    actual_result = get_seeds(test_image_array, test_mask_array)
+    actual_result = get_seeds(test_image_array, test_mask_array, exclusion_mask)
     for index in range(len(expected_result)):
         assert expected_result[index] == actual_result[index]