Improve test robustness

2026-07-27 11:27:08 +08:00 · 2015-05-08 22:26:03 -07:00
parent acc1ee0dad
commit 3b0179e4c1
1 changed files with 2 additions and 2 deletions
@@ -30,13 +30,13 @@ def test_clear_border():

 def test_clear_border_non_binary():
    image = np.array([[1, 2, 3, 1, 2],
-                      [3, 4, 5, 4, 2],
+                      [3, 3, 5, 4, 2],
                      [3, 4, 5, 4, 2],
                      [3, 3, 2, 1, 2]])

    result = clear_border(image.copy())
    expected = np.array([[0, 0, 0, 0, 0],
-                         [0, 4, 5, 4, 0],
+                         [0, 0, 5, 4, 0],
                         [0, 4, 5, 4, 0],
                         [0, 0, 0, 0, 0]])