Coverage for tests/test_NoiseReplacer.py : 41%

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
# This file is part of meas_base. # # Developed for the LSST Data Management System. # This product includes software developed by the LSST Project # (https://www.lsst.org). # See the COPYRIGHT file at the top-level directory of this distribution # for details of code ownership. # # 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 GNU General Public License # along with this program. If not, see <https://www.gnu.org/licenses/>.
"""Plugin that sums instFlux inside and outside the source footprint. """
def getExecutionOrder(): return 2.0
lsst.meas.base.SingleFramePlugin.__init__(self, config, name, schema, metadata) self.insideKey = schema.addField("%s_inside" % (name,), type=np.float64, doc="instFlux inside footprint") self.outsideKey = schema.addField("%s_outside" % (name,), type=np.float64, doc="instFlux outside footprint")
footprint = measRecord.getFootprint() fullArray = exposure.getMaskedImage().getImage().getArray() insideArray = np.zeros(footprint.getArea(), dtype=fullArray.dtype) footprint.spans.flatten(insideArray, fullArray, exposure.getXY0()) insideFlux = float(insideArray.sum()) outsideFlux = float(fullArray.sum()) - insideFlux measRecord.set(self.insideKey, insideFlux) measRecord.set(self.outsideKey, outsideFlux)
self.bbox = lsst.geom.Box2I(lsst.geom.Point2I(-20, -30), lsst.geom.Extent2I(240, 260)) self.dataset = lsst.meas.base.tests.TestDataset(self.bbox) # first source is a point self.dataset.addSource(100000.0, lsst.geom.Point2D(50.1, 49.8)) # second source is extended self.dataset.addSource(120000.0, lsst.geom.Point2D(149.9, 50.3), lsst.afw.geom.Quadrupole(8, 9, 3)) with self.dataset.addBlend() as family: family.addChild(110000.0, lsst.geom.Point2D(65.2, 150.7), lsst.afw.geom.Quadrupole(7, 5, -1)) family.addChild(140000.0, lsst.geom.Point2D(72.3, 149.1)) family.addChild(90000.0, lsst.geom.Point2D(68.5, 156.9))
"""Test noise replacement in single frame measurement.
We compare the instFlux inside and outside source Footprints on an extremely high S/N image. """
# We choose a random seed which causes the test to pass. task = self.makeSingleFrameMeasurementTask("test_NoiseReplacer") exposure, catalog = self.dataset.realize(1.0, task.schema, randomSeed=0) task.run(catalog, exposure) sumVariance = exposure.getMaskedImage().getVariance().getArray().sum() for record in catalog: self.assertFloatsAlmostEqual(record.get("test_NoiseReplacer_inside"), record.get("truth_instFlux"), rtol=1E-3)
# N.B. Next line checks that a random value is correct to a # statistical 1-sigma prediction; some RNG seeds may cause it to # fail (indeed, 67% should) self.assertLess(record.get("test_NoiseReplacer_outside"), np.sqrt(sumVariance))
del self.bbox del self.dataset
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |