diff --git a/tests/test_doubleShapeletPsfApprox.py b/tests/test_doubleShapeletPsfApprox.py index 99fd3d13..1f61dab4 100644 --- a/tests/test_doubleShapeletPsfApprox.py +++ b/tests/test_doubleShapeletPsfApprox.py @@ -180,7 +180,7 @@ def testForcedPlugin(self): config.copyColumns = {"id": "objectId", "parent": "parentObjectId"} self.setupTaskConfig(config) config.slots.centroid = "base_TransformedCentroid" - config.plugins.names |= ["base_TransformedCentroid"] + config.plugins.names |= ["base_TransformedCentroid", "base_PixelFlags"] refSchema = lsst.afw.table.SourceTable.makeMinimalSchema() refCentroidKey = lsst.afw.table.Point2DKey.addFields(refSchema, "centroid", "centroid", "pixel") refSchema.getAliasMap().set("slot_Centroid", "centroid") diff --git a/tests/test_generalShapeletPsfApproxPlugins.py b/tests/test_generalShapeletPsfApproxPlugins.py index eeac0886..0a370ff7 100644 --- a/tests/test_generalShapeletPsfApproxPlugins.py +++ b/tests/test_generalShapeletPsfApproxPlugins.py @@ -110,7 +110,9 @@ def testForced(self): config.slots.modelFlux = None config.doReplaceWithNoise = False config.slots.centroid = "base_TransformedCentroid" - config.plugins.names = ["base_TransformedCentroid", "modelfit_GeneralShapeletPsfApprox"] + config.plugins.names = ["base_TransformedCentroid", + "base_PixelFlags", + "modelfit_GeneralShapeletPsfApprox"] config.plugins["modelfit_GeneralShapeletPsfApprox"].sequence = ["SingleGaussian"] config.copyColumns = {"id": "objectId", "parent": "parentObjectId"} refCat = lsst.afw.table.SourceCatalog(self.schema)