Coverage for tests/test_flaggedSourceSelector.py : 34%

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
#!/usr/bin/env python # # LSST Data Management System # # Copyright 2008-2016 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 <https://www.lsstcorp.org/LegalNotices/>. #
"""Insert a likely-good source into the catalog. Num is added to various values to distinguish them in catalogs with multiple objects. """ record = src.addNew() record['coord_ra'] = (1. + num) * lsst.geom.degrees record['coord_dec'] = (2. + num) * lsst.geom.degrees record['slot_Centroid_x'] = 10. + num record['slot_Centroid_y'] = 20. + num record['slot_ApFlux_instFlux'] = 100. + num record['slot_ApFlux_instFluxErr'] = 1. record.set("calib_psf_used", True)
schema = lsst.meas.base.tests.TestDataset.makeMinimalSchema() schema.addField("slot_ApFlux_instFlux", type=float) schema.addField("slot_ApFlux_instFluxErr", type=float) schema.addField("calib_psf_used", type="Flag")
self.src = afwTable.SourceCatalog(schema) self.sourceSelector = sourceSelector.sourceSelectorRegistry['flagged']()
del self.src del self.sourceSelector
"""Insert good sources and check that they were selected.""" for i in range(5): addGoodSource(self.src, i) result = self.sourceSelector.run(self.src) for x in self.src['id']: self.assertIn(x, result.sourceCat['id'])
"""Add a source that fails the source selector test and check that our output array is indeed empty. """ addGoodSource(self.src, 1) self.src[0].set('calib_psf_used', False) result = self.sourceSelector.run(self.src) self.assertNotIn(self.src['id'][0], result.sourceCat['id'])
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |