diff options
Diffstat (limited to 'unsupported/test/cxx11_tensor_custom_index.cpp')
-rw-r--r-- | unsupported/test/cxx11_tensor_custom_index.cpp | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/unsupported/test/cxx11_tensor_custom_index.cpp b/unsupported/test/cxx11_tensor_custom_index.cpp new file mode 100644 index 000000000..ff9545a7a --- /dev/null +++ b/unsupported/test/cxx11_tensor_custom_index.cpp @@ -0,0 +1,38 @@ +// This file is part of Eigen, a lightweight C++ template library +// for linear algebra. +// +// Copyright (C) 2015 Benoit Steiner <benoit.steiner.goog@gmail.com> +// +// This Source Code Form is subject to the terms of the Mozilla +// Public License v. 2.0. If a copy of the MPL was not distributed +// with this file, You can obtain one at http://mozilla.org/MPL/2.0/. + +#include "main.h" +#include <limits> + +#include <Eigen/Dense> +#include <Eigen/CXX11/Tensor> + +using Eigen::Tensor; + + +template <int DataLayout> +static void test_custom_index() { + + Tensor<float, 4, DataLayout> tensor(2, 3, 5, 7); + tensor.setRandom(); + + using NormalIndex = DSizes<ptrdiff_t, 4>; + using CustomIndex = Matrix<unsigned int , 4, 1>; + CustomIndex coeffC(1,2,4,1); + NormalIndex coeff(1,2,4,1); + + VERIFY_IS_EQUAL(tensor.coeff(coeffC), tensor.coeff(coeff)); + VERIFY_IS_EQUAL(tensor.coeffRef(coeffC), tensor.coeffRef(coeff)); +} + + +void test_cxx11_tensor_custom_index() { + test_custom_index<ColMajor>(); + test_custom_index<RowMajor>(); +} |