def testCosmologyCatalog(self):
     """
     Does a catalog get written?
     """
     dbObj = myTestGals(database=self.dbName)
     cat = cosmologicalGalaxyCatalog(dbObj)
     cat.write_catalog(self.catName)
    def setUp(self):
        self.obs_metadata = ObservationMetaData(mjd=52000.7, bandpassName='i',
                            boundType='circle',unrefractedRA=200.0,unrefractedDec=-30.0,
                            boundLength=1.0, m5 = 25.0)

        self.galaxy = myTestGals(database='PhotometryTestDatabase.db')
        self.star = myTestStars(database='PhotometryTestDatabase.db')
    def setUp(self):
        self.obs_metadata = ObservationMetaData(mjd=52000.7,
                            boundType = 'circle',unrefractedRA=200.0,unrefractedDec=-30.0,
                            boundLength=1.0,
                            m5=[23.9, 25.0, 24.7, 24.0, 23.3, 22.1],
                            bandpassName=['u', 'g', 'r', 'i', 'z', 'y'])

        self.galaxy = myTestGals(database='PhotometryTestDatabase.db')
        self.star = myTestStars(database='PhotometryTestDatabase.db')
    def testCatalogDistanceModulus(self):
        """
        Does cosmologicalDistanceModulus get properly applied
        """
        dbObj = myTestGals(database=self.dbName)
        cosmoCat = cosmologicalGalaxyCatalog(dbObj)
        controlCat = absoluteGalaxyCatalog(dbObj)
        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)
 def setUp(self):
     self.starDB = myTestStars(database=self.starDbName)
     self.galaxyDB = myTestGals(database=self.galaxyDbName)