コード例 #1
0
 def testPersisted(self):
     self.table.definePsfFlux("a")
     self.table.defineCentroid("b")
     self.table.defineShape("c")
     with lsst.utils.tests.getTempFilePath(".fits") as filename:
         self.catalog.writeFits(filename)
         catalog = lsst.afw.table.SourceCatalog.readFits(filename)
         table = catalog.getTable()
         record = catalog[0]
         # I'm using the keys from the non-persisted table.  They should work at least in the
         # current implementation
         self.assertEqual(table.getPsfFluxDefinition(), "a")
         self.assertEqual(record.get(self.fluxKey), record.getPsfFlux())
         self.assertEqual(record.get(self.fluxFlagKey), record.getPsfFluxFlag())
         self.assertEqual(table.getCentroidDefinition(), "b")
         centroid = self.centroidKey.get(self.record)
         self.assertEqual(centroid, record.getCentroid())
         self.assertClose(math.fabs(self.record.get(self.xErrKey)),
             math.sqrt(self.record.getCentroidErr()[0,0]), rtol=1e-6)
         self.assertClose(math.fabs(self.record.get(self.yErrKey)),
             math.sqrt(self.record.getCentroidErr()[1,1]), rtol=1e-6)
         shape = self.shapeKey.get(self.record)
         self.assertEqual(table.getShapeDefinition(), "c")
         self.assertEqual(shape, record.getShape())
         self.assertClose(math.fabs(self.record.get(self.xxErrKey)),
             math.sqrt(self.record.getShapeErr()[0,0]), rtol=1e-6)
         self.assertClose(math.fabs(self.record.get(self.yyErrKey)),
             math.sqrt(self.record.getShapeErr()[1,1]), rtol=1e-6)
         self.assertClose(math.fabs(self.record.get(self.xyErrKey)),
             math.sqrt(self.record.getShapeErr()[2,2]), rtol=1e-6)
コード例 #2
0
 def testPersisted(self):
     self.table.definePsfFlux("a")
     self.table.defineCentroid("b")
     self.table.defineShape("c")
     with lsst.utils.tests.getTempFilePath(".fits") as filename:
         self.catalog.writeFits(filename)
         catalog = lsst.afw.table.SourceCatalog.readFits(filename)
         table = catalog.getTable()
         record = catalog[0]
         # I'm using the keys from the non-persisted table.  They should work at least in the
         # current implementation
         self.assertEqual(table.getPsfFluxDefinition(), "a")
         self.assertEqual(record.get(self.fluxKey), record.getPsfFlux())
         self.assertEqual(record.get(self.fluxFlagKey),
                          record.getPsfFluxFlag())
         self.assertEqual(table.getCentroidDefinition(), "b")
         centroid = self.centroidKey.get(self.record)
         self.assertEqual(centroid, record.getCentroid())
         self.assertFloatsAlmostEqual(
             math.fabs(self.record.get(self.xErrKey)),
             math.sqrt(self.record.getCentroidErr()[0, 0]),
             rtol=1e-6)
         self.assertFloatsAlmostEqual(
             math.fabs(self.record.get(self.yErrKey)),
             math.sqrt(self.record.getCentroidErr()[1, 1]),
             rtol=1e-6)
         shape = self.shapeKey.get(self.record)
         self.assertEqual(table.getShapeDefinition(), "c")
         self.assertEqual(shape, record.getShape())
         self.assertFloatsAlmostEqual(
             math.fabs(self.record.get(self.xxErrKey)),
             math.sqrt(self.record.getShapeErr()[0, 0]),
             rtol=1e-6)
         self.assertFloatsAlmostEqual(
             math.fabs(self.record.get(self.yyErrKey)),
             math.sqrt(self.record.getShapeErr()[1, 1]),
             rtol=1e-6)
         self.assertFloatsAlmostEqual(
             math.fabs(self.record.get(self.xyErrKey)),
             math.sqrt(self.record.getShapeErr()[2, 2]),
             rtol=1e-6)
コード例 #3
0
ファイル: ticketDM-433.py プロジェクト: rnikutta/afw
 def testPersisted(self):
     self.table.definePsfFlux("a")
     self.table.defineCentroid("b")
     self.table.defineShape("c")
     with lsst.utils.tests.getTempFilePath(".fits") as filename:
         self.catalog.writeFits(filename)
         catalog = lsst.afw.table.SourceCatalog.readFits(filename)
         table = catalog.getTable()
         record = catalog[0]
         # I'm using the keys from the non-persisted table.  They should work at least in the
         # current implementation
         self.assertEqual(table.getPsfFluxDefinition(), "a")
         self.assertEqual(record.get(self.fluxKey), record.getPsfFlux())
         self.assertEqual(record.get(self.fluxFlagKey),
                          record.getPsfFluxFlag())
         self.assertEqual(table.getCentroidDefinition(), "b")
         self.assertEqual(record.get(self.centroidKey),
                          record.getCentroid())
         self.assertClose(record.get(self.centroidErrKey),
                          record.getCentroidErr())
         self.assertEqual(table.getShapeDefinition(), "c")
         self.assertEqual(record.get(self.shapeKey), record.getShape())
         self.assertClose(record.get(self.shapeErrKey),
                          record.getShapeErr())