Add test on cutting_plan_method

src/estimation.py

@@ -459,9 +459,11 @@ def __init__(
         self.remove_incoherence()
 
     def __call__(self, x: np.ndarray) -> float:
+        if len(x.shape) > 1:
+            x = x[0]
         return np.maximum(
             self.lower_bound(x), np.sum([layer(x) for layer in self.layers], axis=0)
-        )[0]
+        )
 
     def remove_inconsistence(self, tolerance: float = 1) -> None:
         inputs, costs = self.lower_bound.inputs, self.lower_bound.costs

src/multi_stock_bellman_value_calculation.py

@@ -445,10 +445,18 @@ def Lget_costs(
                         pkl.dump((proto, m.stored_variables_and_constraints_ids), file)
                 costs_ws, slopes_ws, _, _ = get_week_scenario_costs(
                     m=m,
-                    controls_list=controls_list[week, scenario],
+                    controls_list=[
+                        {
+                            a: u[i]
+                            for i, a in enumerate(
+                                multi_stock_management.dict_reservoirs.keys()
+                            )
+                        }
+                        for u in controls_list[week, scenario]
+                    ],
                 )
                 costs[week, scenario] = costs_ws
-                slopes[week, scenario] = slopes_ws
+                slopes[week, scenario] = [[sl for sl in s.values()] for s in slopes_ws]
             with open(filename, "wb") as file:
                 pkl.dump(
                     (controls_list[: week + 1], costs[: week + 1], slopes[: week + 1]),