def testSingleFrameMeasurementTransform(self): """Test applying a transform task to the results of single frame measurement.""" schema = afwTable.SourceTable.makeMinimalSchema() sfmConfig = measBase.SingleFrameMeasurementConfig(plugins=[PLUGIN_NAME]) # We don't use slots in this test for key in sfmConfig.slots: setattr(sfmConfig.slots, key, None) sfmTask = measBase.SingleFrameMeasurementTask(schema, config=sfmConfig) transformTask = TransformTask(measConfig=sfmConfig, inputSchema=sfmTask.schema, outputDataset="src") self._transformAndCheck(sfmConfig, sfmTask.schema, transformTask)
def testForcedMeasurementTransform(self): """Test applying a transform task to the results of forced measurement.""" schema = afwTable.SourceTable.makeMinimalSchema() forcedConfig = measBase.ForcedMeasurementConfig(plugins=[PLUGIN_NAME]) # We don't use slots in this test for key in forcedConfig.slots: setattr(forcedConfig.slots, key, None) forcedConfig.copyColumns = {"id": "objectId", "parent": "parentObjectId"} forcedTask = measBase.ForcedMeasurementTask(schema, config=forcedConfig) transformConfig = TransformConfig(copyFields=("objectId", "coord_ra", "coord_dec")) transformTask = TransformTask(measConfig=forcedConfig, inputSchema=forcedTask.schema, outputDataset="forced_src", config=transformConfig) self._transformAndCheck(forcedConfig, forcedTask.schema, transformTask)