Coverage for tests/test_maskNans.py : 33%

Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
# # LSST Data Management System # Copyright 2012-2013 LSST Corporation. # # This product includes software developed by the # LSST Project (http://www.lsst.org/). # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the LSST License Statement and # the GNU General Public License along with this program. If not, # see <http://www.lsstcorp.org/LegalNotices/>. #
self.size = 100 self.freqImage = 34 self.freqMask = 12 self.freqVariance = 23 self.afterMask = 2 self.allowMask = 1
image = ImageClass(self.size, self.size) x, y = np.indices((self.size, self.size)) image.getImage().getArray()[y, x] = np.where(x*y % self.freqImage, 0, np.nan) image.getMask().getArray()[y, x] = np.where(x*y % self.freqMask, 0, self.allowMask) image.getVariance().getArray()[y, x] = np.where(x*y % self.freqVariance, 0, np.nan)
if debug: ds9.mtv(image.getImage(), frame=1, title="Image") ds9.mtv(image.getVariance(), frame=2, title="Variance") ds9.mtv(image.getMask(), frame=3, title="Original mask")
isUnmasked = np.logical_not(image.getMask().getArray() & self.allowMask) isNan = np.logical_or(np.isnan(image.getImage().getArray()), np.isnan(image.getVariance().getArray()))
maskExpected = np.where(np.logical_and(isUnmasked, isNan), self.afterMask, image.getMask().getArray()) numExpected = np.count_nonzero(maskExpected & self.afterMask)
numNans = maskNans(image, self.afterMask, self.allowMask)
if debug: ds9.mtv(image.getMask(), frame=4, title="UNC-ed mask")
self.assertEqual(numNans, numExpected) self.assertMasksEqual(image.getMask(), maskExpected)
self.check(afwImage.MaskedImageF)
self.check(afwImage.MaskedImageD)
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |