Coverage for tests/test_config.py : 38%

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 pipe_base. # # Developed for the LSST Data Management System. # This product includes software developed by the LSST Project # (http://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 <http://www.gnu.org/licenses/>.
"""
doc="Resource configuration")
doc="Resource configuration")
self.resources.minMemoryMB = 1024 self.resources.minNumCores = 32
dimensions=["UnitA"], storageClass="SCA", doc="") dimensions=["UnitA", "UnitB"], storageClass="SCB", scalar=True, doc="") dimensions=["UnitB", "UnitC"], storageClass="SCC", scalar=False, doc="") storageClass="SCX", doc="") storageClass="SCY", doc="")
"""A test case for Task """
def setUpClass(cls): for name in ("SCA", "SCB", "SCC", "SCX", "SCY"): StorageClassFactory().registerStorageClass(StorageClass(name))
"""Test for a task without resource config """ task = NoResourceTask() res_config = task.getResourceConfig() self.assertIs(res_config, None)
"""Test for a task with resource config """ task = OneTask() res_config = task.getResourceConfig() self.assertIsNot(res_config, None) self.assertIs(res_config.minMemoryMB, None) self.assertEqual(res_config.minNumCores, 1)
"""Test for a task with resource config and special defaults """ task = TwoTask() res_config = task.getResourceConfig() self.assertIsNot(res_config, None) self.assertEqual(res_config.minMemoryMB, 1024) self.assertEqual(res_config.minNumCores, 32)
"""Test for a config without datasets """ config = pexConfig.Config() self.assertEqual(pipeBase.PipelineTask.getInputDatasetTypes(config), {}) self.assertEqual(pipeBase.PipelineTask.getOutputDatasetTypes(config), {}) self.assertEqual(pipeBase.PipelineTask.getInitInputDatasetTypes(config), {}) self.assertEqual(pipeBase.PipelineTask.getInitOutputDatasetTypes(config), {})
"""Test for a config with datasets """ config = ConfigWithDatasets()
descriptors = pipeBase.PipelineTask.getInputDatasetTypes(config) self.assertCountEqual(descriptors.keys(), ["input1", "input2"]) descriptor = descriptors["input1"] self.assertEqual(descriptor.datasetType.name, config.input1.name) self.assertCountEqual(descriptor.datasetType.dimensions.names, config.input1.dimensions) self.assertEqual(descriptor.datasetType.storageClass.name, config.input1.storageClass) self.assertFalse(descriptor.scalar) descriptor = descriptors["input2"] self.assertEqual(descriptor.datasetType.name, config.input2.name) self.assertCountEqual(descriptor.datasetType.dimensions.names, config.input2.dimensions) self.assertEqual(descriptor.datasetType.storageClass.name, config.input2.storageClass) self.assertTrue(descriptor.scalar)
descriptors = pipeBase.PipelineTask.getOutputDatasetTypes(config) self.assertCountEqual(descriptors.keys(), ["output"]) descriptor = descriptors["output"] self.assertEqual(descriptor.datasetType.name, config.output.name) self.assertCountEqual(descriptor.datasetType.dimensions.names, config.output.dimensions) self.assertEqual(descriptor.datasetType.storageClass.name, config.output.storageClass) self.assertFalse(descriptor.scalar)
descriptors = pipeBase.PipelineTask.getInitInputDatasetTypes(config) self.assertCountEqual(descriptors.keys(), ["initInput"]) descriptor = descriptors["initInput"] self.assertEqual(descriptor.datasetType.name, config.initInput.name) self.assertEqual(len(descriptor.datasetType.dimensions), 0) self.assertEqual(descriptor.datasetType.storageClass.name, config.initInput.storageClass) self.assertTrue(descriptor.scalar)
descriptors = pipeBase.PipelineTask.getInitOutputDatasetTypes(config) self.assertCountEqual(descriptors.keys(), ["initOutput"]) descriptor = descriptors["initOutput"] self.assertEqual(descriptor.datasetType.name, config.initOutput.name) self.assertEqual(len(descriptor.datasetType.dimensions), 0) self.assertEqual(descriptor.datasetType.storageClass.name, config.initOutput.storageClass) self.assertTrue(descriptor.scalar)
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |