Merge branch 'tickets/DM-37727'

lsst · Nov 30, 2023 · 2de8f1a · 2de8f1a
2 parents 6f1216f + aadb198
commit 2de8f1a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_schema.py b/tests/test_schema.py
@@ -50,9 +50,8 @@ def _validateSchema(self, dataset, dataId, tableName):
         self.assertEqual(len(sdmSchema), 1)
         expectedColumnNames = set(column['name'] 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, expectedColumnNames)
 
     def testObjectSchemaMatch(self):