Coverage for tests/test_PeakCentroid.py : 50%

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 2008-2017 AURA/LSST. # # 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/>. #
SingleFramePluginTransformSetupHelper, ForcedPluginTransformSetupHelper)
def onlyLogFatal(log): """ For the duration of the context, only log FATAL errors.
This is convenient when testing algorithms under failure conditions: we want to be able to check that they have set appropriate flags without spewing alarming & confusing error messages to the console. """ oldLevel = log.getLevel() log.setLevel(log.FATAL) try: yield finally: log.setLevel(oldLevel)
self.center = lsst.geom.Point2D(50.1, 49.8) self.bbox = lsst.geom.Box2I(lsst.geom.Point2I(-20, -30), lsst.geom.Extent2I(140, 160)) self.dataset = lsst.meas.base.tests.TestDataset(self.bbox) self.dataset.addSource(1000000.0, self.center) self.task = self.makeSingleFrameMeasurementTask("base_PeakCentroid") self.exposure, self.catalog = self.dataset.realize(10.0, self.task.schema, randomSeed=0)
del self.center del self.bbox del self.dataset del self.task del self.exposure del self.catalog
"""Check that we recover the correct location of the centroid.""" self.task.run(self.catalog, self.exposure) x = self.catalog[0].get("base_PeakCentroid_x") y = self.catalog[0].get("base_PeakCentroid_y") self.assertFalse(self.catalog[0].get("base_PeakCentroid_flag")) self.assertFloatsAlmostEqual(x, self.center.getX(), atol=None, rtol=.02) self.assertFloatsAlmostEqual(y, self.center.getY(), atol=None, rtol=.02)
""" When it is impossible to measure the centroid -- in this case, because we have removed the Peaks from the SourceRecord -- the centroider should set a failure flag. """ self.catalog[0].getFootprint().getPeaks().clear() # The decorator suppresses alarming but expected errors on the console. with onlyLogFatal(self.task.log): self.task.run(self.catalog, self.exposure) self.assertTrue(self.catalog[0].get("base_PeakCentroid_flag"))
SingleFramePluginTransformSetupHelper, lsst.utils.tests.TestCase):
""" Helper class to sub in an empty PropertyList as the final argument to lsst.meas.base.SingleFramePeakCentroidPlugin. """
return lsst.meas.base.SingleFramePeakCentroidPlugin(control, name, inputSchema, lsst.daf.base.PropertyList())
ForcedPluginTransformSetupHelper, lsst.utils.tests.TestCase):
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |