testing main for coverage

watertap/examples/flowsheets/oaro/oaro.py

@@ -67,7 +67,7 @@ def erd_type_not_found(erd_type):
     )
 
 
-def main(erd_type=ERDtype.pump_as_turbine):
+def main(erd_type=ERDtype.pump_as_turbine, raise_on_failure=False):
     # set up solver
     solver = get_solver()
 
@@ -77,6 +77,12 @@ def main(erd_type=ERDtype.pump_as_turbine):
     initialize_system(m, solver=solver)
 
     results = solve(m, solver=solver)
+    if not check_optimal_termination(results):
+        msg = "Simulation did not converge"
+        if raise_on_failure:
+            raise RuntimeError(msg)
+        else:
+            print(f"WARNING: {msg}")
 
     print("\n***---Simulation results---***")
     display_system(m)

watertap/examples/flowsheets/oaro/tests/test_oaro.py

@@ -29,6 +29,7 @@
 
 from watertap.unit_models.pressure_changer import Pump, EnergyRecoveryDevice
 from watertap.examples.flowsheets.oaro.oaro import (
+    main,
     build,
     set_operating_conditions,
     initialize_system,
@@ -118,3 +119,7 @@ def test_solution(self, system_frame):
     def test_config_error(self, system_frame):
         with pytest.raises(Exception):
             build(erd_type="not_a_configuration")
+
+    @pytest.mark.component
+    def test_main(self):
+        main(raise_on_failure=True)