From 9961ef14837283d20ecefbf664f8d4d2439203a7 Mon Sep 17 00:00:00 2001 From: jcarpent <jcarpent@laas.fr> Date: Wed, 2 May 2018 17:41:56 +0200 Subject: [PATCH] [Test] Increase ref test with Eigen::Ref example --- unittest/ref.cpp | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/unittest/ref.cpp b/unittest/ref.cpp index 0ca38fc0..a8bff02f 100644 --- a/unittest/ref.cpp +++ b/unittest/ref.cpp @@ -36,12 +36,20 @@ void printVector(const eigenpy::Ref<MatType> & mat) printMatrix(mat); } -template<typename MatType> -void setOnes(eigenpy::Ref<MatType> mat) +template<typename MatType,int Options, typename StrideType> +void setOnes(Eigen::Ref<MatType,Options,StrideType> mat) { mat.setOnes(); } +template<typename MatType> +void setOnes_wrap(eigenpy::Ref<MatType> mat) +{ + setOnes(mat); +} + + + BOOST_PYTHON_MODULE(ref) { namespace bp = boost::python; @@ -53,7 +61,7 @@ BOOST_PYTHON_MODULE(ref) bp::def("printVector", printVector<VectorXd>); - bp::def("setOnes", setOnes<Vector3d>); - bp::def("setOnes", setOnes<VectorXd>); - bp::def("setOnes", setOnes<MatrixXd>); + bp::def("setOnes", setOnes_wrap<Vector3d>); + bp::def("setOnes", setOnes_wrap<VectorXd>); + bp::def("setOnes", setOnes_wrap<MatrixXd>); } -- GitLab