Fix a deprecation warning coming from afw changes

lsst · Jan 14, 2020 · 17e54a1 · 17e54a1
1 parent 283f753
commit 17e54a1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/datasetsHelper.py b/tests/datasetsHelper.py
@@ -43,15 +43,17 @@ def assertCatalogEqual(self, inputCatalog, outputCatalog):
         outputRecord = outputCatalog[0]
         self.assertEqual(inputRecord.getPsfInstFlux(), outputRecord.getPsfInstFlux())
         self.assertEqual(inputRecord.getPsfFluxFlag(), outputRecord.getPsfFluxFlag())
-        self.assertEqual(inputTable.getCentroidDefinition(), outputTable.getCentroidDefinition())
+        self.assertEqual(inputTable.getSchema().getAliasMap().get("slot_Centroid"),
+                         outputTable.getSchema().getAliasMap().get("slot_Centroid"))
         self.assertEqual(inputRecord.getCentroid(), outputRecord.getCentroid())
         self.assertFloatsAlmostEqual(
             inputRecord.getCentroidErr()[0, 0],
             outputRecord.getCentroidErr()[0, 0], rtol=1e-6)
         self.assertFloatsAlmostEqual(
             inputRecord.getCentroidErr()[1, 1],
             outputRecord.getCentroidErr()[1, 1], rtol=1e-6)
-        self.assertEqual(inputTable.getShapeDefinition(), outputTable.getShapeDefinition())
+        self.assertEqual(inputTable.getSchema().getAliasMap().get("slot_Shape"),
+                         outputTable.getSchema().getAliasMap().get("slot_Shape"))
         self.assertFloatsAlmostEqual(
             inputRecord.getShapeErr()[0, 0],
             outputRecord.getShapeErr()[0, 0], rtol=1e-6)