diff options
Diffstat (limited to 'src/boost/libs/multi_array/test/reshape.cpp')
-rw-r--r-- | src/boost/libs/multi_array/test/reshape.cpp | 97 |
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(); +} + + + + + + + |