Coverage for tests/test_metrics.py : 26%

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)
result = self.task.run([]) 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)
catalog = _makeDummyCatalog(3) nCcds = 3 inputData = {"sources": [catalog] * nCcds} inputDataIds = {"sources": [{"visit": 42, "ccd": x} for x in range(nCcds)]} outputDataId = {"measurement": {"visit": 42}} measDirect = self.task.run([catalog]).measurement measMany = self.task.adaptArgsAndRun(inputData, inputDataIds, outputDataId).measurement
assert_quantity_allclose(measMany.quantity, nCcds * 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)
result = self.task.run([], []) 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)
sciCatalog = _makeDummyCatalog(5) diaCatalog = _makeDummyCatalog(1) nCcds = 3 inputData = {"sciSources": [sciCatalog] * nCcds, "diaSources": [diaCatalog] * nCcds} dataIds = [{"visit": 42, "ccd": x} for x in range(nCcds)] inputDataIds = {"sciSources": dataIds, "diaSources": dataIds} outputDataId = {"measurement": {"visit": 42}} measDirect = self.task.run([sciCatalog], [diaCatalog]).measurement measMany = self.task.adaptArgsAndRun(inputData, inputDataIds, outputDataId).measurement
assert_quantity_allclose(measMany.quantity, measDirect.quantity)
# Hack around unittest's hacky test setup system
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |