Move adding fields to test data schema

lsst · Aug 18, 2023 · ad5fbd4 · ad5fbd4
1 parent f61f28a
commit ad5fbd4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python/lsst/meas/base/tests.py b/python/lsst/meas/base/tests.py
@@ -200,6 +200,7 @@ def makeMinimalSchema(cls):
             cls.keys["centroid"] = lsst.afw.table.Point2DKey.addFields(
                 schema, "truth", "true simulated centroid", "pixel"
             )
+            print('coord_raErr' in schema, 'truth_xErr' in schema)
             cls.keys["centroid_sigma"] = lsst.afw.table.CovarianceMatrix2fKey.addFields(
                 schema, "truth", ['x', 'y'], "pixel"
             )
@@ -210,7 +211,6 @@ def makeMinimalSchema(cls):
             )
             cls.keys["isStar"] = schema.addField("truth_isStar", type="Flag",
                                                  doc="set if the object is a star")
-            cls.keys["coord_sigma"] = lsst.afw.table.CoordKey.addErrorFields(schema)
             schema.getAliasMap().set("slot_Shape", "truth")
             schema.getAliasMap().set("slot_Centroid", "truth")
             schema.getAliasMap().set("slot_ModelFlux", "truth")

diff --git a/tests/test_forcedPhot.py b/tests/test_forcedPhot.py
@@ -31,6 +31,7 @@
 
 import lsst.afw.image
 from lsst.afw.math import ChebyshevBoundedField
+from lsst.afw.table import CoordKey
 from lsst.meas.base import ForcedPhotCcdTask, ForcedPhotCcdFromDataFrameTask
 import lsst.meas.base.tests
 import lsst.utils.tests
@@ -50,6 +51,8 @@ def setUp(self):
         dataset.addSource(instFlux=10000, centroid=lsst.geom.Point2D(60, 70))
 
         schema = dataset.makeMinimalSchema()
+        # Add coordinate error fields needed by updateSourceCoords below:
+        CoordKey.addErrorFields(schema)
         self.exposure, self.refCat = dataset.realize(noise=10, schema=schema)
         lsst.afw.table.updateSourceCoords(self.exposure.wcs, self.refCat)
         # Simple aperture correction map in case the task needs it.