From ceadff6ca7ab6211f7d1b31f51418e10f58fc7ac Mon Sep 17 00:00:00 2001 From: peterdudfield Date: Mon, 12 Aug 2024 16:31:50 +0100 Subject: [PATCH] fix test --- src/tests/test_elexon_forecast.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tests/test_elexon_forecast.py b/src/tests/test_elexon_forecast.py index 46aa61a..7721420 100644 --- 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