147 lines
4.1 KiB
C++
147 lines
4.1 KiB
C++
// Copyright (c) 2018-2019 Cem Bassoy
|
|
//
|
|
// Distributed under 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)
|
|
//
|
|
// The authors gratefully acknowledge the support of
|
|
// Fraunhofer and Google in producing this work
|
|
// which started as a Google Summer of Code project.
|
|
//
|
|
|
|
#include <iostream>
|
|
#include <algorithm>
|
|
#include <complex>
|
|
#include <boost/numeric/ublas/tensor.hpp>
|
|
#include <boost/numeric/ublas/tensor/multi_index.hpp>
|
|
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
#include "utility.hpp"
|
|
|
|
|
|
BOOST_AUTO_TEST_SUITE ( test_multi_index )
|
|
|
|
|
|
using test_types = zip<int,long,float,double,std::complex<float>>::with_t<boost::numeric::ublas::first_order, boost::numeric::ublas::last_order>;
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE ( test_index_classes )
|
|
{
|
|
using namespace boost::numeric::ublas::index;
|
|
|
|
|
|
BOOST_CHECK_EQUAL ( _a.value , 1 ) ;
|
|
BOOST_CHECK_EQUAL ( _b.value , 2 ) ;
|
|
BOOST_CHECK_EQUAL ( _c.value , 3 ) ;
|
|
BOOST_CHECK_EQUAL ( _d.value , 4 ) ;
|
|
BOOST_CHECK_EQUAL ( _e.value , 5 ) ;
|
|
BOOST_CHECK_EQUAL ( _f.value , 6 ) ;
|
|
BOOST_CHECK_EQUAL ( _g.value , 7 ) ;
|
|
BOOST_CHECK_EQUAL ( _h.value , 8 ) ;
|
|
BOOST_CHECK_EQUAL ( _i.value , 9 ) ;
|
|
BOOST_CHECK_EQUAL ( _j.value , 10 ) ;
|
|
BOOST_CHECK_EQUAL ( _k.value , 11 ) ;
|
|
BOOST_CHECK_EQUAL ( _l.value , 12 ) ;
|
|
BOOST_CHECK_EQUAL ( _m.value , 13 ) ;
|
|
BOOST_CHECK_EQUAL ( _n.value , 14 ) ;
|
|
BOOST_CHECK_EQUAL ( _o.value , 15 ) ;
|
|
BOOST_CHECK_EQUAL ( _p.value , 16 ) ;
|
|
BOOST_CHECK_EQUAL ( _q.value , 17 ) ;
|
|
BOOST_CHECK_EQUAL ( _r.value , 18 ) ;
|
|
BOOST_CHECK_EQUAL ( _s.value , 19 ) ;
|
|
BOOST_CHECK_EQUAL ( _t.value , 20 ) ;
|
|
BOOST_CHECK_EQUAL ( _u.value , 21 ) ;
|
|
BOOST_CHECK_EQUAL ( _v.value , 22 ) ;
|
|
BOOST_CHECK_EQUAL ( _w.value , 23 ) ;
|
|
BOOST_CHECK_EQUAL ( _x.value , 24 ) ;
|
|
BOOST_CHECK_EQUAL ( _y.value , 25 ) ;
|
|
BOOST_CHECK_EQUAL ( _z.value , 26 ) ;
|
|
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE ( test_multi_index_class_construction )
|
|
{
|
|
using namespace boost::numeric::ublas;
|
|
using namespace boost::numeric::ublas::index;
|
|
|
|
|
|
{
|
|
multi_index<2> ind(_a, _b);
|
|
|
|
BOOST_CHECK_EQUAL ( get<0>( ind ), 1 ) ;
|
|
BOOST_CHECK_EQUAL ( get<1>( ind ), 2 ) ;
|
|
}
|
|
|
|
|
|
{
|
|
multi_index<2> ind(_d,_c);
|
|
|
|
BOOST_CHECK_EQUAL ( ind[0] , 4 ) ;
|
|
BOOST_CHECK_EQUAL ( ind[1] , 3 ) ;
|
|
}
|
|
}
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE_TEMPLATE( test_tensor_multi_index_class_generation, value, test_types )
|
|
{
|
|
using namespace boost::numeric::ublas;
|
|
using value_type = typename value::first_type;
|
|
using layout_type = typename value::second_type;
|
|
using tensor_type = tensor<value_type,layout_type>;
|
|
|
|
auto t = std::make_tuple (
|
|
index::_a, // 0
|
|
index::_b, // 1
|
|
index::_c, // 2
|
|
index::_d, // 3
|
|
index::_e // 4
|
|
);
|
|
|
|
{
|
|
auto a = tensor_type(shape{2,3}, value_type{2});
|
|
auto a_ind = a( std::get<0>(t), std::get<2>(t) );
|
|
|
|
BOOST_CHECK_EQUAL ( std::addressof( a_ind.first ), std::addressof( a ) ) ;
|
|
|
|
BOOST_CHECK_EQUAL (std::get<0>(a_ind.second)(), index::_a() ) ;
|
|
BOOST_CHECK_EQUAL (std::get<1>(a_ind.second)(), index::_c() ) ;
|
|
}
|
|
|
|
{
|
|
auto a = tensor_type(shape{2,3}, value_type{2});
|
|
auto a_ind = a( std::get<2>(t), std::get<0>(t) );
|
|
|
|
BOOST_CHECK_EQUAL ( std::addressof( a_ind.first ), std::addressof( a ) ) ;
|
|
|
|
BOOST_CHECK_EQUAL (std::get<0>(a_ind.second)(), index::_c() ) ;
|
|
BOOST_CHECK_EQUAL (std::get<1>(a_ind.second)(), index::_a() ) ;
|
|
}
|
|
|
|
{
|
|
auto a = tensor_type(shape{2,3}, value_type{2});
|
|
auto a_ind = a( std::get<2>(t), std::get<3>(t) );
|
|
|
|
BOOST_CHECK_EQUAL (std::addressof( a_ind.first ), std::addressof( a ) ) ;
|
|
|
|
BOOST_CHECK_EQUAL (std::get<0>(a_ind.second)(), index::_c() ) ;
|
|
BOOST_CHECK_EQUAL (std::get<1>(a_ind.second)(), index::_d() ) ;
|
|
}
|
|
|
|
{
|
|
auto a = tensor_type(shape{2,3,4}, value_type{2});
|
|
auto a_ind = a( std::get<2>(t), std::get<3>(t), std::get<0>(t) );
|
|
|
|
BOOST_CHECK_EQUAL (std::addressof( a_ind.first ), std::addressof( a ) ) ;
|
|
|
|
BOOST_CHECK_EQUAL (std::get<0>(a_ind.second)(), index::_c() ) ;
|
|
BOOST_CHECK_EQUAL (std::get<1>(a_ind.second)(), index::_d() ) ;
|
|
BOOST_CHECK_EQUAL (std::get<2>(a_ind.second)(), index::_a() ) ;
|
|
}
|
|
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|
|
|