Add has_xxx and __contains__ to Python SummaryState

2020-01-07 15:13:12 +01:00
parent 9ae007ddc9
commit b9c3df5f42
2 changed files with 8 additions and 1 deletions
--- a/python/cxx/summary_state.cpp
+++ b/python/cxx/summary_state.cpp
@@ -51,5 +51,8 @@ void python::common::export_SummaryState(py::module& module) {
        .def("elapsed", &SummaryState::get_elapsed)
        .def_property_readonly("groups", groups)
        .def_property_readonly("wells", wells)
+        .def("__contains__", &SummaryState::has)
+        .def("has_well_var", &SummaryState::has_well_var)
+        .def("has_group_var", &SummaryState::has_group_var)
        .def("__getitem__", &SummaryState::get);
 }