fix test

openclimatefix · Aug 12, 2024 · ceadff6 · ceadff6
1 parent 8401d81
commit ceadff6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/tests/test_elexon_forecast.py b/src/tests/test_elexon_forecast.py
@@ -9,7 +9,7 @@
 
 class MockClass(BaseModel):
 
-    publish_time: str
+    start_time: str
     quantity: float
     business_type: Optional[str] = "Solar generation"
 
@@ -52,7 +52,7 @@ def test_get_elexon_forecast_mock(mock_function, api_client):
     assert len(api_data) == len(mock_data)
     for i in range(len(api_data)):
         assert api_data[i]["expected_power_generation_megawatts"] == mock_data[i].quantity
-        assert pd.Timestamp(api_data[i]["timestamp"]) == pd.Timestamp(mock_data[i].publish_time)
+        assert pd.Timestamp(api_data[i]["timestamp"]) == pd.Timestamp(mock_data[i].start_time)
 
 
 @pytest.mark.integration