fix

Author: peterdudfield

nowcasting_datamodel/update.py

@@ -116,14 +116,16 @@ def update_all_forecast_latest(forecasts: List[ForecastSQL], session: Session):
     """
 
     # get all latest forecasts
-    forecasts_historic = get_latest_forecast_for_gsps(session=session, historic=True)
+    forecasts_historic_all_gsps = get_latest_forecast_for_gsps(session=session, historic=True)
 
     for forecast in forecasts:
 
         # chose the correct forecast historic
         gsp_id = forecast.location.gsp_id
         forecast_historic = [
-            forecast for forecast in forecasts_historic if forecast.location.gsp_id == gsp_id
+            forecast_historic_one_gsps
+            for forecast_historic_one_gsps in forecasts_historic_all_gsps
+            if forecast.location.gsp_id == gsp_id
         ]
         if len(forecast_historic) == 0:
             forecast_historic = None

tests/test_update.py

@@ -44,9 +44,11 @@ def test_update_one_gsp(db_session):
 
     update_forecast_latest(forecast=forecast_sql, session=db_session)
 
+    forecast_values_latest = db_session.query(ForecastValueLatestSQL).all()
     assert len(db_session.query(ForecastValueSQL).all()) == 2
-    assert len(db_session.query(ForecastValueLatestSQL).all()) == 2
+    assert len(forecast_values_latest) == 2
     assert len(db_session.query(ForecastSQL).all()) == 2
+    assert forecast_values_latest[0].expected_power_generation_megawatts == 2
 
     # new forecast is made
     # create and add