[unittest] Test returning reference of std::pair

stack-of-tasks · Sep 18, 2024 · 9b01d56 · 9b01d56
1 parent 694d996
commit 9b01d56
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/unittest/python/test_std_pair.py b/unittest/python/test_std_pair.py
@@ -1,5 +1,6 @@
-from std_pair import copy, std_pair_to_tuple
+from std_pair import copy, passthrough, std_pair_to_tuple
 
 t = (1, 2.0)
 assert std_pair_to_tuple(t) == t
 assert copy(t) == t
+assert passthrough(t) == t
diff --git a/unittest/std_pair.cpp b/unittest/std_pair.cpp
@@ -3,7 +3,6 @@
 
 #include <eigenpy/eigenpy.hpp>
 #include <eigenpy/std-pair.hpp>
-#include <iostream>
 
 namespace bp = boost::python;
 
@@ -17,6 +16,11 @@ std::pair<T1, T2> copy(const std::pair<T1, T2>& pair) {
   return pair;
 }
 
+template <typename T1, typename T2>
+const std::pair<T1, T2>& passthrough(const std::pair<T1, T2>& pair) {
+  return pair;
+}
+
 BOOST_PYTHON_MODULE(std_pair) {
   eigenpy::enableEigenPy();
 
@@ -25,4 +29,6 @@ BOOST_PYTHON_MODULE(std_pair) {
 
   bp::def("std_pair_to_tuple", std_pair_to_tuple<int, double>);
   bp::def("copy", copy<int, double>);
+  bp::def("passthrough", passthrough<int, double>,
+          bp::return_value_policy<bp::copy_const_reference>());
 }