Fix diff with mask

camptocamp · Dec 22, 2023 · 915ba99 · 915ba99
1 parent 7004a3b
commit 915ba99
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/c2cwsgiutils/acceptance/image.py b/c2cwsgiutils/acceptance/image.py
@@ -148,6 +148,7 @@ def check_image(
         assert (
             expected.shape[0] == mask.shape[0] and expected.shape[1] == mask.shape[1]
         ), f"Mask and expected image should have the same shape ({mask.shape} != {expected.shape})"
+        expected[mask] = [255, 255, 255]
 
     assert (
         expected.shape == image_to_check.shape