Merge branch 'main' into sc_voltage_results_parameter

powsybl · Sep 22, 2023 · 7ee4bdc · 7ee4bdc
2 parents d8efa29 + 3029ed8
commit 7ee4bdc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/cpp/src/bindings.cpp b/cpp/src/bindings.cpp
@@ -307,8 +307,11 @@ PYBIND11_MODULE(_pypowsybl, m) {
             .value("CONVERGED", pypowsybl::LoadFlowComponentStatus::CONVERGED, "The loadflow has converged.")
             .value("FAILED", pypowsybl::LoadFlowComponentStatus::FAILED, "The loadflow has failed.")
             .value("MAX_ITERATION_REACHED", pypowsybl::LoadFlowComponentStatus::MAX_ITERATION_REACHED, "The loadflow has reached its maximum iterations count.")
-            .value("SOLVER_FAILED", pypowsybl::LoadFlowComponentStatus::SOLVER_FAILED, "The loadflow numerical solver has failed.");
-
+            .value("SOLVER_FAILED", pypowsybl::LoadFlowComponentStatus::SOLVER_FAILED, "The loadflow numerical solver has failed.")
+            .def("__bool__", [](const pypowsybl::LoadFlowComponentStatus& status) {
+                return status == pypowsybl::LoadFlowComponentStatus::CONVERGED;
+            });
+
     py::enum_<pypowsybl::PostContingencyComputationStatus>(m, "PostContingencyComputationStatus", "Loadflow status for one connected component after contingency for security analysis.")
             .value("CONVERGED", pypowsybl::PostContingencyComputationStatus::CONVERGED, "The loadflow has converged.")
             .value("FAILED", pypowsybl::PostContingencyComputationStatus::FAILED, "The loadflow has failed.")

diff --git a/tests/test_loadflow.py b/tests/test_loadflow.py
@@ -268,3 +268,9 @@ def test_run_lf_with_report():
     pp.loadflow.run_ac(n2, reporter=reporter)
     report3 = str(reporter)
     assert len(report3) > len(report2)
+
+
+def test_result_status_as_bool():
+    n = pp.network.create_ieee14()
+    r = pp.loadflow.run_ac(n)
+    assert r[0].status