Coverage for tests/testCosmologyMixins.py : 44%

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
lsst.utils.tests.init()
'uBulge', 'gBulge', 'rBulge', 'iBulge', 'zBulge', 'yBulge', 'uDisk', 'gDisk', 'rDisk', 'iDisk', 'zDisk', 'yDisk', 'uAgn', 'gAgn', 'rAgn', 'iAgn', 'zAgn', 'yAgn', 'redshift', 'cosmologicalDistanceModulus']
'uBulge', 'gBulge', 'rBulge', 'iBulge', 'zBulge', 'yBulge', 'uDisk', 'gDisk', 'rDisk', 'iDisk', 'zDisk', 'yDisk', 'uAgn', 'gAgn', 'rAgn', 'iAgn', 'zAgn', 'yAgn', 'redshift']
""" Must set this to zero rather than `None` so that PhotometryGalaxies does not apply cosmological dimming """ return np.zeros(len(self.column_by_name('galid')))
""" This class will test to make sure that our example CosmologyMixin (defined in lsst/sims/photUtils/examples/CosmologyMixin.py) can produce a catalog """
def setUpClass(cls): cls.dbName = tempfile.mktemp(prefix='cosmologyTestDB-', suffix=".db", dir=ROOT) cls.dbSize = 100 if os.path.exists(cls.dbName): os.unlink(cls.dbName) makeGalTestDB(size=cls.dbSize, seedVal=1, filename=cls.dbName)
def tearDownClass(cls): sims_clean_up() if os.path.exists(cls.dbName): os.unlink(cls.dbName)
del cls.dbName del cls.dbSize
self.obs = ObservationMetaData(mjd=59580.0)
""" Does a catalog get written? """ dbObj = myTestGals(database=self.dbName) cat = cosmologicalGalaxyCatalog(dbObj, obs_metadata=self.obs) with lsst.utils.tests.getTempFilePath('.txt') as catName: cat.write_catalog(catName)
""" Does cosmologicalDistanceModulus get properly applied """ dbObj = myTestGals(database=self.dbName) cosmoCat = cosmologicalGalaxyCatalog(dbObj, obs_metadata=self.obs) controlCat = absoluteGalaxyCatalog(dbObj, obs_metadata=self.obs) cosmoIter = cosmoCat.iter_catalog(chunk_size=self.dbSize) controlIter = controlCat.iter_catalog(chunk_size=self.dbSize)
cosmology = CosmologyObject()
for (cosmoRow, controlRow) in zip(cosmoIter, controlIter): modulus = cosmology.distanceModulus(controlRow[25]) self.assertEqual(cosmoRow[0], controlRow[0]) self.assertEqual(cosmoRow[25], controlRow[25]) self.assertEqual(cosmoRow[26], modulus) for i in range(1, 25): self.assertAlmostEqual(cosmoRow[i], controlRow[i] + modulus, 6)
lsst.utils.tests.init() unittest.main() |