Fix test

NREL · May 29, 2024 · 14c794b · 14c794b
1 parent f2b8136
commit 14c794b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_bespoke.py b/tests/test_bespoke.py
@@ -723,7 +723,7 @@ def test_collect_bespoke():
         collector.collect("cf_profile-2012")
 
         with Resource(h5_file) as fout:
-            meta = fout.meta
+            meta = fout.meta.rename(columns=SupplyCurveField.map_from_legacy())
             assert all(
                 meta[SupplyCurveField.GID].values
                 == sorted(meta[SupplyCurveField.GID].values)
@@ -737,8 +737,7 @@ def test_collect_bespoke():
         for fp in source_fps:
             with Resource(fp) as source:
                 src_meta = source.meta.rename(
-                    SupplyCurveField.map_from_legacy(), axis=1
-                )
+                    columns=SupplyCurveField.map_from_legacy())
                 assert all(
                     np.isin(
                         src_meta[SupplyCurveField.GID].values,