summaryrefslogtreecommitdiffstats
path: root/src/boost/libs/multi_array/test/reshape.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/boost/libs/multi_array/test/reshape.cpp')
-rw-r--r--src/boost/libs/multi_array/test/reshape.cpp97
1 files changed, 97 insertions, 0 deletions
diff --git a/src/boost/libs/multi_array/test/reshape.cpp b/src/boost/libs/multi_array/test/reshape.cpp
new file mode 100644
index 000000000..a611fe427
--- /dev/null
+++ b/src/boost/libs/multi_array/test/reshape.cpp
@@ -0,0 +1,97 @@
+// Copyright 2002 The Trustees of Indiana University.
+
+// Use, modification and distribution is subject to the Boost Software
+// License, Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
+// http://www.boost.org/LICENSE_1_0.txt)
+
+// Boost.MultiArray Library
+// Authors: Ronald Garcia
+// Jeremy Siek
+// Andrew Lumsdaine
+// See http://www.boost.org/libs/multi_array for documentation.
+
+//
+// reshape.cpp - testing reshaping functionality
+//
+
+#include <boost/multi_array.hpp>
+
+#include <boost/core/lightweight_test.hpp>
+
+#include <boost/array.hpp>
+#include <boost/type.hpp>
+
+int
+main()
+{
+ const int ndims=3;
+ typedef boost::multi_array<int,ndims> array;
+ typedef boost::multi_array_ref<int,ndims> array_ref;
+ typedef boost::const_multi_array_ref<int,ndims> const_array_ref;
+
+ boost::array<array::size_type,ndims> dims = {{2,3,4}};
+ boost::array<array::size_type,ndims> new_dims = {{4,3,2}};
+
+ int data[] = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,
+ 14,15,16,17,18,19,20,21,22,23};
+ const int data_size=24;
+
+ // Basic reshape test
+ {
+ array A(dims);
+ A.assign(data,data+data_size);
+
+ array_ref B(data,dims);
+ const_array_ref C(data,dims);
+
+ A.reshape(new_dims);
+ B.reshape(new_dims);
+ C.reshape(new_dims);
+
+ int* ptr = data;
+ for (array::index i = 0; i != 4; ++i)
+ for (array::index j = 0; j != 3; ++j)
+ for (array::index k = 0; k != 2; ++k) {
+ BOOST_TEST(A[i][j][k] == *ptr);
+ BOOST_TEST(B[i][j][k] == *ptr);
+ BOOST_TEST(C[i][j][k] == *ptr++);
+ }
+ }
+
+ // Ensure that index bases are preserved over reshape
+ {
+ boost::array<array::index,ndims> bases = {{0, 1, -1}};
+
+ array A(dims);
+ A.assign(data,data+data_size);
+
+ array_ref B(data,dims);
+ const_array_ref C(data,dims);
+
+ A.reindex(bases);
+ B.reindex(bases);
+ C.reindex(bases);
+
+ A.reshape(new_dims);
+ B.reshape(new_dims);
+ C.reshape(new_dims);
+
+ int* ptr = data;
+ for (array::index i = 0; i != 4; ++i)
+ for (array::index j = 1; j != 4; ++j)
+ for (array::index k = -1; k != 1; ++k) {
+ BOOST_TEST(A[i][j][k] == *ptr);
+ BOOST_TEST(B[i][j][k] == *ptr);
+ BOOST_TEST(C[i][j][k] == *ptr++);
+ }
+ }
+
+ return boost::report_errors();
+}
+
+
+
+
+
+
+