update background subtraction test with clip

slaclab · Dec 2, 2024 · d30acfe · d30acfe
1 parent 0f9099d
commit d30acfe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit_tests/lcls_tools/common/image/test_processing.py b/tests/unit_tests/lcls_tools/common/image/test_processing.py
@@ -53,7 +53,7 @@ def test_subtract_background(self):
         image_processor = ImageProcessor(background_image=background_image)
         image = image_processor.subtract_background(self.image)
         image = image
-        background = (self.image - 1)
+        background = np.clip(self.image - 1, 0, None)
         np.testing.assert_array_equal(
             image, background,
             err_msg=("expected image to equal background "
@@ -67,7 +67,7 @@ def test_subtract_background(self):
         image_processor = ImageProcessor(threshold=1)
         image = image_processor.subtract_background(self.image)
         image = image
-        background = (self.image - 1)
+        background = np.clip(self.image - 1, 0, None)
         np.testing.assert_array_equal(
             image, background,
             err_msg=("expected image to equal background "