From facc81d7412aca74e4d4f48b880a7943def3e41b Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Mon, 2 Mar 2026 14:45:51 +0000 Subject: [PATCH] expanded keys being checked in tests --- backend/export/tests/test_export.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/backend/export/tests/test_export.py b/backend/export/tests/test_export.py index be3c5dae..72eafbe5 100644 --- a/backend/export/tests/test_export.py +++ b/backend/export/tests/test_export.py @@ -248,3 +248,17 @@ def test_default_export_integration(db_session): # Check the houses assert failed.shape[0] + + assert df["total_retrofit_cost"].sum() == 41706.585999999996, ( + "Expected total retrofit cost to be 10000, got {}".format(df["total_retrofit_cost"].sum()) + ) + + assert df["predicted_post_works_sap"].sum() == 698.1, ( + "Expected total predicted post works SAP to be 698.1, got {}".format(df["predicted_post_works_sap"].sum()) + ) + + assert df["sap_points"].sum() == 100.10000000000001, ( + "Expected total SAP points increase to be 100.10000000000001, got {}".format(df["sap_points"].sum()) + ) + + assert df.shape == (10, 95), "Expected dataframe shape to be (10, 11), got {}".format(df.shape)