123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146 |
- // 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()
|