fix test

openclimatefix · Dec 19, 2023 · 6b52128 · 6b52128
1 parent 3edd3bc
commit 6b52128
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/tests/eval/test_metrics.py b/tests/eval/test_metrics.py
@@ -9,12 +9,22 @@ def test_metrics():
 
     results_df = pd.DataFrame(
         columns=[
-            "id",
+            "pv_id",
             "timestamp",
             "horizon_hour",
             "forecast_power",
             "generation_power",
-        ], data=np.random.random((100,5)))
+        ],
+        data=np.random.random((100, 5)),
+    )
+
+    pv_metadata = pd.DataFrame(
+        columns=[
+            "pv_id",
+            "capacity",
+        ],
+        data=np.random.random((100, 2)),
+    )
 
     # call the metrics function
-    metrics(results_df)
+    metrics(results_df, pv_metadata)