test/ref: test for const Eigen::Ref<const ...

Author: jcarpent

unittest/eigen_ref.cpp

@@ -1,10 +1,11 @@
 /*
  * Copyright 2014-2019, CNRS
- * Copyright 2018-2020, INRIA
+ * Copyright 2018-2022, INRIA
  */
 
 #include <iostream>
 
+#include "eigenpy/eigen-from-python.hpp"
 #include "eigenpy/eigenpy.hpp"
 
 using namespace Eigen;
@@ -41,6 +42,16 @@ Eigen::Ref<MatType> asRef(const int rows, const int cols) {
   return mat;
 }
 
+template <typename MatType>
+Eigen::Ref<MatType> asRef(Eigen::Ref<MatType> mat) {
+  return Eigen::Ref<MatType>(mat);
+}
+
+template <typename MatType>
+const Eigen::Ref<const MatType> asConstRef(Eigen::Ref<MatType> mat) {
+  return Eigen::Ref<const MatType>(mat);
+}
+
 BOOST_PYTHON_MODULE(eigen_ref) {
   namespace bp = boost::python;
   eigenpy::enableEigenPy();
@@ -60,5 +71,10 @@ BOOST_PYTHON_MODULE(eigen_ref) {
   bp::def("fillVec", fill<VectorXd>);
   bp::def("fill", fill<MatrixXd>);
 
-  bp::def("asRef", asRef<MatrixXd>);
+  bp::def("asRef",
+          (Eigen::Ref<MatrixXd>(*)(const int, const int))asRef<MatrixXd>);
+  bp::def("asRef",
+          (Eigen::Ref<MatrixXd>(*)(Eigen::Ref<MatrixXd>))asRef<MatrixXd>);
+  bp::def("asConstRef", (const Eigen::Ref<const MatrixXd> (*)(
+                            Eigen::Ref<MatrixXd>))asConstRef<MatrixXd>);
 }

unittest/python/test_eigen_ref.py

@@ -17,10 +17,15 @@ def test(mat):
   A_ref2.fill(0)
   assert np.array_equal(A_ref,A_ref2)
 
+  ref = asRef(mat)
+  assert np.all(ref == mat)
+
+  const_ref = asConstRef(mat)
+  assert np.all(const_ref == mat)
 
 rows = 10
 cols = 30
 
-mat = np.array(np.zeros((rows,cols)))
+mat = np.ones((rows,cols),order='F')
 
 test(mat)