Skip to content

Commit

Permalink
Specify the storage class when reading in the table so that it
Browse files Browse the repository at this point in the history
doesn't randomly pick between astropy tables and pandas.
  • Loading branch information
sr525 committed Dec 1, 2023
1 parent c4c265b commit d151d16
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_schema.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ def _validateSchema(self, dataset, dataId, tableName):
expectedColumns = {column['name']: column['datatype']
for column in sdmSchema[0]['columns']}

df = self.butler.get(dataset, dataId)
df = self.butler.get(dataset, dataId, storageClass="DataFrame")
df.reset_index(inplace=True)
outputColumnNames = set(df.columns.to_list())
self.assertEqual(outputColumnNames, set(expectedColumns.keys()), f"{info} failed")
Expand Down

0 comments on commit d151d16

Please sign in to comment.