def testColumnOperators(self):
""" Test operator overloading on columns """
columnSlice = afwImage.ImageSliceF(self.imgProjectCol.getImage())
imgAdd = self.img + columnSlice
imgAdd2 = columnSlice + self.img
imgSub = self.img - columnSlice
imgMul = self.img * columnSlice
imgMul2 = columnSlice * self.img
imgDiv = self.img / columnSlice
for i in range(self.n):
self.assertAlmostEqual(
imgAdd[0, i, afwImage.LOCAL], self.img[0, i, afwImage.LOCAL] +
columnSlice[0, i, afwImage.LOCAL])
self.assertAlmostEqual(imgAdd2[0, i, afwImage.LOCAL],
imgAdd[0, i, afwImage.LOCAL])
self.assertAlmostEqual(
imgSub[0, i, afwImage.LOCAL], self.img[0, i, afwImage.LOCAL] -
columnSlice[0, i, afwImage.LOCAL])
self.assertAlmostEqual(
imgMul[0, i, afwImage.LOCAL], self.img[0, i, afwImage.LOCAL] *
columnSlice[0, i, afwImage.LOCAL])
self.assertAlmostEqual(imgMul2[0, i, afwImage.LOCAL],
imgMul[0, i, afwImage.LOCAL])
self.assertAlmostEqual(
imgDiv[0, i, afwImage.LOCAL], self.img[0, i, afwImage.LOCAL] /
columnSlice[0, i, afwImage.LOCAL])
def testColumnOperators(self):
""" Test operator overloading on columns """
columnSlice = afwImage.ImageSliceF(self.imgProjectCol.getImage())
imgAdd = self.img + columnSlice
imgAdd2 = columnSlice + self.img
imgSub = self.img - columnSlice
imgMul = self.img * columnSlice
imgMul2 = columnSlice * self.img
imgDiv = self.img / columnSlice
for i in range(self.n):
self.assertAlmostEqual(imgAdd.get(0, i),
self.img.get(0, i) + columnSlice.get(0, i))
self.assertAlmostEqual(imgAdd2.get(0, i), imgAdd.get(0, i))
self.assertAlmostEqual(imgSub.get(0, i),
self.img.get(0, i) - columnSlice.get(0, i))
self.assertAlmostEqual(imgMul.get(0, i),
self.img.get(0, i) * columnSlice.get(0, i))
self.assertAlmostEqual(imgMul2.get(0, i), imgMul.get(0, i))
self.assertAlmostEqual(imgDiv.get(0, i),
self.img.get(0, i) / columnSlice.get(0, i))