Fixing missing inputs to treat runtime params method

davidusb-geek · Dec 25, 2024 · fd31007 · fd31007
1 parent 5ca132f
commit fd31007
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 1 deletion.
diff --git a/src/emhass/forecast.py b/src/emhass/forecast.py
@@ -956,7 +956,7 @@ def get_load_forecast(
             if self.get_data_from_file:
                 filename_path = self.emhass_conf["data_path"] / "test_df_final.pkl"
                 with open(filename_path, "rb") as inp:
-                    rh.df_final, days_list, var_list = pickle.load(inp)
+                    rh.df_final, days_list, var_list, rh.ha_config = pickle.load(inp)
                     self.var_load = var_list[0]
                     self.retrieve_hass_conf["sensor_power_load_no_var_loads"] = (
                         self.var_load

diff --git a/tests/test_command_line_utils.py b/tests/test_command_line_utils.py
@@ -149,6 +149,7 @@ def test_set_input_data_dict(self):
             "load_power_forecast": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
             "load_cost_forecast": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
             "prod_price_forecast": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
+            "prediction_horizon": 10,
         }
         runtimeparams_json = json.dumps(runtimeparams)
         params = copy.deepcopy(json.loads(self.params_json))

diff --git a/tests/test_forecast.py b/tests/test_forecast.py
@@ -416,6 +416,7 @@ def test_get_forecasts_with_lists(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
         # Build RetrieveHass Object
         rh = RetrieveHass(
@@ -560,6 +561,7 @@ def test_get_forecasts_with_longer_lists(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
         # Create Forecast Object
         fcst = Forecast(
@@ -653,6 +655,7 @@ def test_get_forecasts_with_lists_special_case(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
         # Create RetrieveHass Object
         rh = RetrieveHass(

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -162,6 +162,7 @@ def test_treat_runtimeparams(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
         self.assertIsInstance(params, str)
         params = json.loads(params)
@@ -184,6 +185,7 @@ def test_treat_runtimeparams(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
         self.assertIsInstance(params, str)
         params = json.loads(params)
@@ -249,6 +251,7 @@ def test_treat_runtimeparams(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
         self.assertIsInstance(params, str)
         params = json.loads(params)
@@ -344,6 +347,7 @@ def test_treat_runtimeparams_failed(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
 
         self.assertTrue(
@@ -402,6 +406,7 @@ def test_treat_runtimeparams_failed(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
         self.assertIsInstance(runtimeparams["pv_power_forecast"], list)
         self.assertIsInstance(runtimeparams["load_power_forecast"], list)
@@ -433,6 +438,7 @@ def test_treat_runtimeparams_failed(self):
             set_type,
             logger,
             emhass_conf,
+            {},
         )
         self.assertIsInstance(runtimeparams["pv_power_forecast"], str)
         self.assertIsInstance(runtimeparams["load_power_forecast"], str)