From f8736d3574707bb38b7c234a91faa41b35472ef8 Mon Sep 17 00:00:00 2001
From: Jun-te Kim <juntekim@googlemail.com>
Date: Tue, 31 Mar 2026 11:42:20 +0000
Subject: [PATCH] added sql model to db

---
 .github/workflows/unit_tests.yml     | 11 +++++++++++
 sfr/principal_pitch/2_export_data.py | 10 ++++------
 2 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/unit_tests.yml b/.github/workflows/unit_tests.yml
index 740f88f7..436428f9 100644
--- a/.github/workflows/unit_tests.yml
+++ b/.github/workflows/unit_tests.yml
@@ -33,6 +33,17 @@ jobs:
       - name: Build test image
         run: docker build -f Dockerfile.test -t model-test .
 
+      - name: Initialise database schema
+        run: |
+          docker run --rm \
+            --network host \
+            -e DB_HOST=localhost \
+            -e DB_NAME=test \
+            -e DB_USERNAME=test \
+            -e DB_PASSWORD=test \
+            -e DB_PORT=5432 \
+            model-test python scripts/init_db.py
+
       - name: Run tests
         run: |
           docker run --rm \
diff --git a/sfr/principal_pitch/2_export_data.py b/sfr/principal_pitch/2_export_data.py
index df54749e..c89560cb 100644
--- a/sfr/principal_pitch/2_export_data.py
+++ b/sfr/principal_pitch/2_export_data.py
@@ -26,15 +26,13 @@ from backend.app.db.functions.materials_functions import get_materials
 from collections import defaultdict
 from sqlalchemy import func
 
-# PORTFOLIO_ID = 206
-# SCENARIOS = [389]
-PORTFOLIO_ID = 633
-SCENARIOS = [1146]
+PORTFOLIO_ID = 639
+SCENARIOS = [1157]
 scenario_names = {
-    1146: "Most Economic",
+    1157: "EPC C - no EWI solid floor",
 }
 
-project_name = "WCHG EPC D rated properties"
+project_name = "Instagroup Sample"
 
 
 def get_data(portfolio_id, scenario_ids):