test: test to-python for Eigen::Ref

Author: jcarpent

unittest/eigen_ref.cpp

@@ -37,6 +37,14 @@ void fill(Eigen::Ref<MatType> mat, const typename MatType::Scalar & value)
   mat.fill(value);
 }
 
+template<typename MatType>
+Eigen::Ref<MatType> asRef(const int rows, const int cols)
+{
+  static MatType mat(rows,cols);
+  std::cout << "mat:\n" << mat << std::endl;
+  return mat;
+}
+
 BOOST_PYTHON_MODULE(eigen_ref)
 {
   namespace bp = boost::python;
@@ -56,4 +64,6 @@ BOOST_PYTHON_MODULE(eigen_ref)
   bp::def("fillVec3", fill<Vector3d>);
   bp::def("fillVec", fill<VectorXd>);
   bp::def("fill", fill<MatrixXd>);
+  
+  bp::def("asRef", asRef<MatrixXd>);
 }

unittest/python/test_eigen_ref.py

@@ -8,6 +8,16 @@ def test(mat):
   printMatrix(mat)
   assert np.array_equal(mat,np.full(mat.shape,1.))
 
+  A_ref = asRef(mat.shape[0],mat.shape[1])
+  A_ref.fill(1.)
+  A_ref2 = asRef(mat.shape[0],mat.shape[1])
+
+  assert np.array_equal(A_ref,A_ref2)
+
+  A_ref2.fill(0)
+  assert np.array_equal(A_ref,A_ref2)
+  
+
 rows = 10
 cols = 30