def testSlotUndefine(self): """Test that we can correctly define and undefine a slot after a SourceTable has been created""" schema = lsst.afw.table.SourceTable.makeMinimalSchema() key = schema.addField("a_flux", type=float, doc="flux field") table = lsst.afw.table.SourceTable.make(schema) table.definePsfFlux("a") self.assertEqual(table.getPsfFluxKey(), key) table.schema.getAliasMap().erase("slot_PsfFlux") self.assertFalse(table.getPsfFluxKey().isValid())
def testSlotUndefine(self): """Test that we can correctly define and undefine a slot after a SourceTable has been created""" schema = lsst.afw.table.SourceTable.makeMinimalSchema() key = schema.addField("a_flux", type=float, doc="flux field") table = lsst.afw.table.SourceTable.make(schema) table.definePsfFlux("a") self.assertEqual(table.getPsfFluxKey(), key) table.schema.getAliasMap().erase("slot_PsfFlux") self.assertFalse(table.getPsfFluxKey().isValid())