1493 Bindings for get_times() function (#1494)

HenrZu · web-flow · commit a4a7b77ade5f · 2026-02-23T15:34:46.000+01:00
Add bindings for the get_times() function from the TimeSeries class
diff --git a/pycode/memilio-simulation/memilio/simulation/bindings/utils/time_series.cpp b/pycode/memilio-simulation/memilio/simulation/bindings/utils/time_series.cpp
@@ -112,7 +112,13 @@ void bind_time_series(py::module_& m, std::string const& name)
                                                                      self.get_num_time_points());
                 return Eigen::Ref<mio::TimeSeries<double>::Matrix>(m);
             },
-            py::return_value_policy::reference_internal);
+            py::return_value_policy::reference_internal)
+
+        .def("get_times", [](const mio::TimeSeries<double>& self) {
+            auto times_range = self.get_times();
+            std::vector<double> times(times_range.begin(), times_range.end());
+            return times;
+        });
 }
 
 } // namespace pymio
diff --git a/pycode/memilio-simulation/tests/test_time_series.py b/pycode/memilio-simulation/tests/test_time_series.py
@@ -69,6 +69,14 @@ def test_print_table(self):
         self.assertEqual(
             output, '\nTime a \n2.00 1.00\n3.50 2.00\n')
 
+    def test_get_times(self):
+        """Test get_times method"""
+        ts = mio.TimeSeries(1)
+        ts.add_time_point(1.5, np.r_[10.0])
+        ts.add_time_point(2.5, np.r_[20.0])
+        times = ts.get_times()
+        self.assertEqual(times, [1.5, 2.5])
+
     def test_print_table_with_separator(self):
         """Test print_table with custom separator"""
         ts = mio.TimeSeries(1)
