From 9036cda36484c4d7268b928b5976534c8ef3ce42 Mon Sep 17 00:00:00 2001 From: Gael Guennebaud Date: Sun, 29 Jan 2017 14:57:45 +0100 Subject: Cleanup intitial reshape implementation: - reshape -> reshaped - make it compatible with evaluators. --- test/reshape.cpp | 43 ++++++++++++++++++++++--------------------- 1 file changed, 22 insertions(+), 21 deletions(-) (limited to 'test/reshape.cpp') diff --git a/test/reshape.cpp b/test/reshape.cpp index 0298a2fe4..e2c045aa6 100644 --- a/test/reshape.cpp +++ b/test/reshape.cpp @@ -15,35 +15,36 @@ using Eigen::MatrixXi; // just test a 4x4 matrix, enumerate all combination manually, // so I don't have to do template-meta-programming here. template -void reshape_all_size(MatType m) { +void reshape_all_size(MatType m) +{ typedef Eigen::Map MapMat; // dynamic - VERIFY_IS_EQUAL((m.template reshape( 1, 16)), MapMat(m.data(), 1, 16)); - VERIFY_IS_EQUAL((m.template reshape( 2, 8)), MapMat(m.data(), 2, 8)); - VERIFY_IS_EQUAL((m.template reshape( 4, 4)), MapMat(m.data(), 4, 4)); - VERIFY_IS_EQUAL((m.template reshape( 8, 2)), MapMat(m.data(), 8, 2)); - VERIFY_IS_EQUAL((m.template reshape(16, 1)), MapMat(m.data(), 16, 1)); + VERIFY_IS_EQUAL((m.reshaped( 1, 16)), MapMat(m.data(), 1, 16)); + VERIFY_IS_EQUAL((m.reshaped( 2, 8)), MapMat(m.data(), 2, 8)); + VERIFY_IS_EQUAL((m.reshaped( 4, 4)), MapMat(m.data(), 4, 4)); + VERIFY_IS_EQUAL((m.reshaped( 8, 2)), MapMat(m.data(), 8, 2)); + VERIFY_IS_EQUAL((m.reshaped(16, 1)), MapMat(m.data(), 16, 1)); // static - VERIFY_IS_EQUAL((m.template reshape< 1, 16>()), MapMat(m.data(), 1, 16)); - VERIFY_IS_EQUAL((m.template reshape< 2, 8>()), MapMat(m.data(), 2, 8)); - VERIFY_IS_EQUAL((m.template reshape< 4, 4>()), MapMat(m.data(), 4, 4)); - VERIFY_IS_EQUAL((m.template reshape< 8, 2>()), MapMat(m.data(), 8, 2)); - VERIFY_IS_EQUAL((m.template reshape<16, 1>()), MapMat(m.data(), 16, 1)); + VERIFY_IS_EQUAL((m.template reshaped< 1, 16>()), MapMat(m.data(), 1, 16)); + VERIFY_IS_EQUAL((m.template reshaped< 2, 8>()), MapMat(m.data(), 2, 8)); + VERIFY_IS_EQUAL((m.template reshaped< 4, 4>()), MapMat(m.data(), 4, 4)); + VERIFY_IS_EQUAL((m.template reshaped< 8, 2>()), MapMat(m.data(), 8, 2)); + VERIFY_IS_EQUAL((m.template reshaped<16, 1>()), MapMat(m.data(), 16, 1)); // reshape chain VERIFY_IS_EQUAL( (m - .template reshape( 1, 16) - .template reshape< 2, 8>() - .template reshape(16, 1) - .template reshape< 8, 2>() - .template reshape( 2, 8) - .template reshape< 1, 16>() - .template reshape( 4, 4) - .template reshape<16, 1>() - .template reshape( 8, 2) - .template reshape< 4, 4>() + . reshaped( 1, 16) + .template reshaped< 2, 8>() + . reshaped(16, 1) + .template reshaped< 8, 2>() + . reshaped( 2, 8) + .template reshaped< 1, 16>() + . reshaped( 4, 4) + .template reshaped<16, 1>() + . reshaped( 8, 2) + .template reshaped< 4, 4>() ), MapMat(m.data(), 4, 4) ); -- cgit v1.2.3