Coverage for tests/test_metrics.py : 31%

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 ip_diffim. # # 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/>.
NumberSciSourcesMetricTask, \ FractionDiaSourcesToSciSourcesMetricTask
catalog = SourceCatalog(SourceCatalog.Table.makeMinimalSchema()) for i in range(size): catalog.addNew() return catalog
def makeTask(cls): return NumberSciSourcesMetricTask()
catalog = _makeDummyCatalog(3) result = self.task.run(catalog) meas = result.measurement
self.assertEqual(meas.metric_name, Name(metric="ip_diffim.numSciSources")) self.assertEqual(meas.quantity, len(catalog) * u.count)
catalog = _makeDummyCatalog(0) result = self.task.run(catalog) meas = result.measurement
self.assertEqual(meas.metric_name, Name(metric="ip_diffim.numSciSources")) self.assertEqual(meas.quantity, 0 * u.count)
result = self.task.run(None) meas = result.measurement self.assertIsNone(meas)
config = self.taskClass.ConfigClass() types = self.taskClass.getInputDatasetTypes(config) # dict.keys() is a collections.abc.Set, which has a narrower interface than __builtins__.set... self.assertSetEqual(set(types.keys()), {"sources"}) self.assertEqual(types["sources"], "src")
catalog = _makeDummyCatalog(3) inputData = {"sources": catalog} inputDataIds = {"sources": {"visit": 42, "ccd": 1}} outputDataId = {"measurement": {"visit": 42, "ccd": 1}} measDirect = self.task.run(catalog).measurement measIndirect = self.task.adaptArgsAndRun(inputData, inputDataIds, outputDataId).measurement
assert_quantity_allclose(measIndirect.quantity, measDirect.quantity)
def makeTask(cls): return FractionDiaSourcesToSciSourcesMetricTask()
sciCatalog = _makeDummyCatalog(5) diaCatalog = _makeDummyCatalog(3) result = self.task.run(sciCatalog, diaCatalog) meas = result.measurement
self.assertEqual(meas.metric_name, Name(metric="ip_diffim.fracDiaSourcesToSciSources")) self.assertEqual(meas.quantity, len(diaCatalog) / len(sciCatalog) * u.dimensionless_unscaled)
sciCatalog = _makeDummyCatalog(5) diaCatalog = _makeDummyCatalog(0) result = self.task.run(sciCatalog, diaCatalog) meas = result.measurement
self.assertEqual(meas.metric_name, Name(metric="ip_diffim.fracDiaSourcesToSciSources")) self.assertEqual(meas.quantity, 0.0 * u.dimensionless_unscaled)
sciCatalog = _makeDummyCatalog(0) diaCatalog = _makeDummyCatalog(3) with self.assertRaises(MetricComputationError): self.task.run(sciCatalog, diaCatalog)
sciCatalog = _makeDummyCatalog(0) diaCatalog = _makeDummyCatalog(0) with self.assertRaises(MetricComputationError): self.task.run(sciCatalog, diaCatalog)
result = self.task.run(None, None) meas = result.measurement self.assertIsNone(meas)
result = self.task.run(sciSources=_makeDummyCatalog(3), diaSources=None) meas = result.measurement self.assertIsNone(meas)
config = self.taskClass.ConfigClass() types = self.taskClass.getInputDatasetTypes(config) # dict.keys() is a collections.abc.Set, which has a narrower interface than __builtins__.set... self.assertSetEqual(set(types.keys()), {"sciSources", "diaSources"}) self.assertEqual(types["sciSources"], "src") self.assertEqual(types["diaSources"], "deepDiff_diaSrc")
# TODO: add a test for the templating in FractionDiaSourcesToSciSourcesMetricConfig # once we've migrated to PipelineTaskConfig
sciCatalog = _makeDummyCatalog(5) diaCatalog = _makeDummyCatalog(1) inputData = {"sciSources": sciCatalog, "diaSources": diaCatalog} inputDataIds = {"sciSources": {"visit": 42, "ccd": 1}, "diaSources": {"visit": 42, "ccd": 1}} outputDataId = {"measurement": {"visit": 42, "ccd": 1}} measDirect = self.task.run(sciCatalog, diaCatalog).measurement measIndirect = self.task.adaptArgsAndRun(inputData, inputDataIds, outputDataId).measurement
assert_quantity_allclose(measIndirect.quantity, measDirect.quantity)
# Hack around unittest's hacky test setup system
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |