From d287df1b824083b892f4970a3d3c20bef38efdf1 Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Fri, 27 Mar 2026 01:46:58 +0000 Subject: [PATCH] hard coded buckets --- backend/tests/test_rebaselining_pipeline.py | 11 ++++++++++- pytest.ini | 2 ++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/backend/tests/test_rebaselining_pipeline.py b/backend/tests/test_rebaselining_pipeline.py index 03a87352..9fbe1f35 100644 --- a/backend/tests/test_rebaselining_pipeline.py +++ b/backend/tests/test_rebaselining_pipeline.py @@ -83,7 +83,16 @@ def test_rebaselining_pipeline_with_real_data(): model_api = ModelApi( portfolio_id="test-portfolio", timestamp=datetime.now().isoformat(), - prediction_buckets=get_prediction_buckets(), + prediction_buckets={ + "sap_change_predictions": "retrofit-sap-predictions-dev", + "heat_demand_predictions": "retrofit-heat-predictions-dev", + "carbon_change_predictions": "retrofit-carbon-predictions-dev", + "heating_kwh_predictions": "retrofit-heating-kwh-predictions-dev", + "hotwater_kwh_predictions": "retrofit-hotwater-kwh-predictions-dev", + "retrofit_sap_baseline_predictions": "retrofit-sap-baseline-predictions-dev", + "retrofit_carbon_baseline_predictions": "retrofit-carbon-baseline-predictions-dev", + "retrofit_heat_baseline_predictions": "retrofit-heat-baseline-predictions-dev", + }, max_retries=1 ) bucket = "retrofit-data-dev" diff --git a/pytest.ini b/pytest.ini index 608d5e0c..ecb17089 100644 --- a/pytest.ini +++ b/pytest.ini @@ -4,3 +4,5 @@ log_cli = true log_cli_level = INFO addopts = --cov-report term-missing --cov=etl/epc --cov=recommendations --cov=backend --cov=etl/epc_clean --cov=etl/spatial testpaths = recommendations/tests backend/tests etl/epc/tests etl/epc_clean/tests etl/spatial/tests backend/condition/tests backend/address2UPRN/tests backend/onboarders/tests backend/categorisation/tests backend/export/tests +markers = + integration: mark a test as an integration test