add Optimal status assertion on compute_cost

Giovanni Amorim · Giovanni Amorim · commit 06229d6c7bf8 · 2025-02-09T18:38:04.000-03:00
diff --git a/src/simulation.jl b/src/simulation.jl
@@ -13,12 +13,14 @@ function compute_single_step_cost(
         yhat
     )
     optimize!(model.plan)
+    @assert termination_status(model.plan) == MOI.OPTIMAL "Optimization failed for PLAN model"
     fix.(assess_forecast_vars(model), y; force=true)
     set_normalized_rhs.(
         model.assess[:assess_policy_fix], 
         value.(plan_policy_vars(model))
     )
     optimize!(model.assess)
+    @assert termination_status(model.assess) == MOI.OPTIMAL "Optimization failed for ASSESS model"
     return objective_value(model.assess)
 end
 

Original file line number	Diff line number	Diff line change
`@@ -13,12 +13,14 @@ function compute_single_step_cost(`
`13`	`13`	`yhat`
`14`	`14`	`)`
`15`	`15`	`optimize!(model.plan)`
	`16`	`+ @assert termination_status(model.plan) == MOI.OPTIMAL "Optimization failed for PLAN model"`
`16`	`17`	`fix.(assess_forecast_vars(model), y; force=true)`
`17`	`18`	`set_normalized_rhs.(`
`18`	`19`	`model.assess[:assess_policy_fix],`
`19`	`20`	`value.(plan_policy_vars(model))`
`20`	`21`	`)`
`21`	`22`	`optimize!(model.assess)`
	`23`	`+ @assert termination_status(model.assess) == MOI.OPTIMAL "Optimization failed for ASSESS model"`
`22`	`24`	`return objective_value(model.assess)`
`23`	`25`	`end`
`24`	`26`