Update schema test to expect and recieve an astropy table

lsst · Dec 1, 2023 · 60069d6 · 60069d6
1 parent 5c27103
commit 60069d6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_schema.py b/tests/test_schema.py
@@ -51,9 +51,8 @@ def _validateSchema(self, dataset, dataId, tableName):
         expectedColumns = {column['name']: column['datatype']
                            for column in sdmSchema[0]['columns']}
 
-        df = self.butler.get(dataset, dataId)
-        df.reset_index(inplace=True)
-        outputColumnNames = set(df.columns.to_list())
+        df = self.butler.get(dataset, dataId, storageClass="ArrowAstropy")
+        outputColumnNames = set(df.colnames)
         self.assertEqual(outputColumnNames, set(expectedColumns.keys()), f"{info} failed")
 
         # the data type mapping from felis datatype to pandas