Coverage for tests/datasetsHelper.py : 23%

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 daf_butler. # # 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/>.
import lsst.afw.table catalogPath = os.path.join(self.testDir, "data", "basic", "source_catalog.fits") return lsst.afw.table.SourceCatalog.readFits(catalogPath)
import lsst.afw.table self.assertIsInstance(outputCatalog, lsst.afw.table.SourceCatalog) inputTable = inputCatalog.getTable() inputRecord = inputCatalog[0] outputTable = outputCatalog.getTable() outputRecord = outputCatalog[0] self.assertEqual(inputRecord.getPsfInstFlux(), outputRecord.getPsfInstFlux()) self.assertEqual(inputRecord.getPsfFluxFlag(), outputRecord.getPsfFluxFlag()) self.assertEqual(inputTable.getCentroidDefinition(), outputTable.getCentroidDefinition()) self.assertEqual(inputRecord.getCentroid(), outputRecord.getCentroid()) self.assertFloatsAlmostEqual( inputRecord.getCentroidErr()[0, 0], outputRecord.getCentroidErr()[0, 0], rtol=1e-6) self.assertFloatsAlmostEqual( inputRecord.getCentroidErr()[1, 1], outputRecord.getCentroidErr()[1, 1], rtol=1e-6) self.assertEqual(inputTable.getShapeDefinition(), outputTable.getShapeDefinition()) self.assertFloatsAlmostEqual( inputRecord.getShapeErr()[0, 0], outputRecord.getShapeErr()[0, 0], rtol=1e-6) self.assertFloatsAlmostEqual( inputRecord.getShapeErr()[1, 1], outputRecord.getShapeErr()[1, 1], rtol=1e-6) self.assertFloatsAlmostEqual( inputRecord.getShapeErr()[2, 2], outputRecord.getShapeErr()[2, 2], rtol=1e-6)
"""Helper methods for Datasets"""
conform=True): """Make a DatasetType and wrap it in a DatasetRef for a test""" datasetType = DatasetType(datasetTypeName, dimensions, storageClass) if id is None: self.id += 1 id = self.id if run is None: run = Run(id=1, collection="dummy") return DatasetRef(datasetType, dataId, id=id, run=run, conform=conform)
"""Helper methods for Datastore tests"""
"""Shared setUp code for all Datastore tests""" self.registry = registryClass()
# Need to keep ID for each datasetRef since we have no butler # for these tests self.id = 1
self.config = configClass(self.configFile)
# Some subclasses override the working root directory if self.root is not None: self.datastoreType.setConfigRoot(self.root, self.config, self.config.copy())
"""Make a new Datastore instance of the appropriate type.
Parameters ---------- sub : str, optional If not None, the returned Datastore will be distinct from any Datastore constructed with a different value of ``sub``. For PosixDatastore, for example, the converse is also true, and ``sub`` is used as a subdirectory to form the new root.
Returns ------- datastore : `Datastore` Datastore constructed by this routine using the supplied optional subdirectory if supported. """ config = self.config.copy() if sub is not None and self.root is not None: self.datastoreType.setConfigRoot(os.path.join(self.root, sub), config, self.config) if sub is not None: # Ensure that each datastore gets its own registry registryClass = type(self.registry) registry = registryClass() else: registry = self.registry return self.datastoreType(config=config, registry=registry)
"""A formatter that never works but does leave a file behind."""
raise NotImplementedError("This formatter can not read anything")
"""Write an empty file and then raise an exception.""" with open(self.fileDescriptor.location.path, "wb"): pass raise RuntimeError("Did not succeed in writing file")
"""A formatter that always fails without writing anything."""
raise RuntimeError("Did not writing anything at all") |