From a8312fca14626c236c1fbd57cd05ee8349fc73af Mon Sep 17 00:00:00 2001 From: Justin Carpentier <justin.carpentier@inria.fr> Date: Wed, 5 Feb 2020 08:39:57 +0100 Subject: [PATCH] test/matrix: fix --- unittest/python/test_matrix.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/unittest/python/test_matrix.py b/unittest/python/test_matrix.py index 41a450c9..18fce68e 100644 --- a/unittest/python/test_matrix.py +++ b/unittest/python/test_matrix.py @@ -62,19 +62,19 @@ assert( np.array_equal(M,eigenpy.reflex(M,verbose)) ); if verbose: print("===> Block Vector 1x0:6:2") M = Mref[1:2,0:6:2] -assert( np.array_equal(M,eigenpy.reflex(M,verbose)) ); +assert( np.array_equal(M.squeeze(),eigenpy.reflex(M,verbose)) ); if verbose: print("===> Block Vector 1x0:6:2 tanspose") M = Mref[1:2,0:6:2].T -assert( np.array_equal(M,eigenpy.reflex(M,verbose)) ); +assert( np.array_equal(M.squeeze(),eigenpy.reflex(M,verbose)) ); if verbose: print("===> Block Vector 0:6:2x1") M = Mref[0:6:2,1:2] -assert( np.array_equal(M,eigenpy.reflex(M,verbose)) ); +assert( np.array_equal(M.squeeze(),eigenpy.reflex(M,verbose)) ); if verbose: print("===> Block Vector 0:6:2x1 tanspose") M = Mref[0:6:2,1:2].T -assert( np.array_equal(M,eigenpy.reflex(M,verbose)) ); +assert( np.array_equal(M.squeeze(),eigenpy.reflex(M,verbose)) ); if verbose: print("===> From Py to Eigen::VectorXd") if verbose: print("===> From Py to Eigen::VectorXd") @@ -87,7 +87,7 @@ M = Mref[0:6:2,1].T if verbose: print("===> Block Vector 0:6:2x1") M = Mref[0:6:2,1:2] -assert( np.array_equal(M,eigenpy.reflexV(M,verbose)) ); +assert( np.array_equal(M.squeeze(),eigenpy.reflexV(M,verbose)) ); if verbose: print("===> Block Vector 0:6:2x1 transpose") M = Mref[0:6:2,1:2].T -- GitLab