From 19a89815278f2b9a4029b726d972d2939455be94 Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Tue, 27 Jun 2023 11:34:13 +0100 Subject: [PATCH] setup test --- model_data/tests/test_floor_recommendations.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/model_data/tests/test_floor_recommendations.py b/model_data/tests/test_floor_recommendations.py index e5238e17..e6a3c012 100644 --- a/model_data/tests/test_floor_recommendations.py +++ b/model_data/tests/test_floor_recommendations.py @@ -4,6 +4,16 @@ import os from unittest.mock import Mock from model_data.recommendations.FloorRecommendations import FloorRecommendations +with open( + os.path.abspath(os.path.dirname(__file__)) + "/model_data/tests/test_data/input_properties.pkl", "rb" +) as f: + input_properties = pickle.load(f) + +with open( + os.path.abspath(os.path.dirname(__file__)) + "/model_data/tests/test_data/uvalue_estimates.pkl", "rb" +) as f: + uvalue_estimates = pickle.load(f) + class TestWallRecommendations: