Add test for addTo and subtractFrom

lsst · Oct 29, 2021 · c495574 · c495574
1 parent 8359f11
commit c495574
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/tests/test_heavyFootprint.py b/tests/test_heavyFootprint.py
@@ -301,6 +301,43 @@ def testDot(self):
             self.assertEqual(hfp1.dot(hfp2), dot)
             self.assertEqual(hfp2.dot(hfp1), dot)
 
+    def testAddSubtract(self):
+        """Test that we can add and subtract a HeavyFootprint."""
+        fs = afwDetect.FootprintSet(self.mi, afwDetect.Threshold(1))
+
+        fs.makeHeavy(self.mi)
+
+        bbox = lsst.geom.BoxI(lsst.geom.PointI(9, 1), lsst.geom.ExtentI(7, 4))
+        compImage = self.mi.Factory(self.mi, bbox, afwImage.LOCAL, True)
+        compImage.set((10, 0x0, 0))
+
+        blankImage = compImage.clone()
+        blankImage.set((0, 0x0, 0))
+
+        testImage = compImage.clone()
+
+        for foot in fs.getFootprints():
+            # Add via addTo
+            foot.addTo(testImage.image)
+
+            # Add by using a temporary image and insert
+            foot.insert(blankImage.image)
+            inserted = blankImage.image.array > 0
+            compImage.image.array[inserted] += blankImage.image.array[inserted]
+
+            np.testing.assert_array_almost_equal(testImage.image.array,
+                                                 compImage.image.array)
+
+            # Subtract via subtractFrom
+            foot.subtractFrom(testImage.image)
+
+            compImage.image.array[inserted] -= blankImage.image.array[inserted]
+
+            np.testing.assert_array_almost_equal(testImage.image.array,
+                                                 compImage.image.array)
+
+            blankImage.set((0, 0x0, 0))
+
 
 class TestMemory(lsst.utils.tests.MemoryTestCase):
     pass