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)