ublas/test/tensor/test_strides.cpp
2019-02-25 17:05:02 +01:00

173 lines
4.9 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 <boost/test/unit_test.hpp>
#include <boost/numeric/ublas/tensor/strides.hpp>
#include <boost/numeric/ublas/tensor/extents.hpp>
//BOOST_AUTO_TEST_SUITE(test_strides, * boost::unit_test::depends_on("test_extents"));
BOOST_AUTO_TEST_SUITE(test_strides)
using test_types = std::tuple<boost::numeric::ublas::first_order, boost::numeric::ublas::last_order>;
BOOST_AUTO_TEST_CASE_TEMPLATE( test_strides_ctor, value, test_types)
{
using namespace boost::numeric;
using extents_type = ublas::basic_extents<unsigned>;
using strides_type = ublas::strides<value>;
strides_type s0{};
BOOST_CHECK ( s0.empty());
BOOST_CHECK_EQUAL ( s0.size(), 0);
strides_type s1{extents_type{1,1}};
BOOST_CHECK (!s1.empty());
BOOST_CHECK_EQUAL ( s1.size(), 2);
strides_type s2{extents_type{1,2}};
BOOST_CHECK (!s2.empty());
BOOST_CHECK_EQUAL ( s2.size(), 2);
strides_type s3{extents_type{2,1}};
BOOST_CHECK (!s3.empty());
BOOST_CHECK_EQUAL ( s3.size(), 2);
strides_type s4{extents_type{2,3}};
BOOST_CHECK (!s4.empty());
BOOST_CHECK_EQUAL ( s4.size(), 2);
strides_type s5{extents_type{2,3,1}};
BOOST_CHECK (!s5.empty());
BOOST_CHECK_EQUAL ( s5.size(), 3);
strides_type s6{extents_type{1,2,3}};
BOOST_CHECK (!s6.empty());
BOOST_CHECK_EQUAL ( s6.size(), 3);
strides_type s7{extents_type{4,2,3}};
BOOST_CHECK (!s7.empty());
BOOST_CHECK_EQUAL ( s7.size(), 3);
}
BOOST_AUTO_TEST_CASE( test_strides_ctor_access_first_order)
{
using namespace boost::numeric;
using extents_type = ublas::basic_extents<unsigned>;
using strides_type = ublas::strides<ublas::first_order>;
strides_type s1{extents_type{1,1}};
BOOST_REQUIRE_EQUAL( s1.size(),2);
BOOST_CHECK_EQUAL ( s1[0], 1);
BOOST_CHECK_EQUAL ( s1[1], 1);
strides_type s2{extents_type{1,2}};
BOOST_REQUIRE_EQUAL ( s2.size(),2);
BOOST_CHECK_EQUAL ( s2[0], 1);
BOOST_CHECK_EQUAL ( s2[1], 1);
strides_type s3{extents_type{2,1}};
BOOST_REQUIRE_EQUAL ( s3.size(),2);
BOOST_CHECK_EQUAL ( s3[0], 1);
BOOST_CHECK_EQUAL ( s3[1], 1);
strides_type s4{extents_type{2,3}};
BOOST_REQUIRE_EQUAL ( s4.size(),2);
BOOST_CHECK_EQUAL ( s4[0], 1);
BOOST_CHECK_EQUAL ( s4[1], 2);
strides_type s5{extents_type{2,3,1}};
BOOST_REQUIRE_EQUAL ( s5.size(),3);
BOOST_CHECK_EQUAL ( s5[0], 1);
BOOST_CHECK_EQUAL ( s5[1], 2);
BOOST_CHECK_EQUAL ( s5[2], 6);
strides_type s6{extents_type{1,2,3}};
BOOST_REQUIRE_EQUAL ( s6.size(),3);
BOOST_CHECK_EQUAL ( s6[0], 1);
BOOST_CHECK_EQUAL ( s6[1], 1);
BOOST_CHECK_EQUAL ( s6[2], 2);
strides_type s7{extents_type{2,1,3}};
BOOST_REQUIRE_EQUAL ( s7.size(),3);
BOOST_CHECK_EQUAL ( s7[0], 1);
BOOST_CHECK_EQUAL ( s7[1], 2);
BOOST_CHECK_EQUAL ( s7[2], 2);
strides_type s8{extents_type{4,2,3}};
BOOST_REQUIRE_EQUAL ( s8.size(),3);
BOOST_CHECK_EQUAL ( s8[0], 1);
BOOST_CHECK_EQUAL ( s8[1], 4);
BOOST_CHECK_EQUAL ( s8[2], 8);
}
BOOST_AUTO_TEST_CASE( test_strides_ctor_access_last_order)
{
using namespace boost::numeric;
using extents_type = ublas::basic_extents<unsigned>;
using strides_type = ublas::strides<ublas::last_order>;
strides_type s1{extents_type{1,1}};
BOOST_REQUIRE_EQUAL( s1.size(),2);
BOOST_CHECK_EQUAL ( s1[0], 1);
BOOST_CHECK_EQUAL ( s1[1], 1);
strides_type s2{extents_type{1,2}};
BOOST_REQUIRE_EQUAL ( s2.size(),2);
BOOST_CHECK_EQUAL ( s2[0], 1);
BOOST_CHECK_EQUAL ( s2[1], 1);
strides_type s3{extents_type{2,1}};
BOOST_REQUIRE_EQUAL ( s3.size(),2);
BOOST_CHECK_EQUAL ( s3[0], 1);
BOOST_CHECK_EQUAL ( s3[1], 1);
strides_type s4{extents_type{2,3}};
BOOST_REQUIRE_EQUAL ( s4.size(),2);
BOOST_CHECK_EQUAL ( s4[0], 3);
BOOST_CHECK_EQUAL ( s4[1], 1);
strides_type s5{extents_type{2,3,1}};
BOOST_REQUIRE_EQUAL ( s5.size(),3);
BOOST_CHECK_EQUAL ( s5[0], 3);
BOOST_CHECK_EQUAL ( s5[1], 1);
BOOST_CHECK_EQUAL ( s5[2], 1);
strides_type s6{extents_type{1,2,3}};
BOOST_REQUIRE_EQUAL ( s6.size(),3);
BOOST_CHECK_EQUAL ( s6[0], 6);
BOOST_CHECK_EQUAL ( s6[1], 3);
BOOST_CHECK_EQUAL ( s6[2], 1);
strides_type s7{extents_type{2,1,3}};
BOOST_REQUIRE_EQUAL ( s7.size(),3);
BOOST_CHECK_EQUAL ( s7[0], 3);
BOOST_CHECK_EQUAL ( s7[1], 3);
BOOST_CHECK_EQUAL ( s7[2], 1);
strides_type s8{extents_type{4,2,3}};
BOOST_REQUIRE_EQUAL ( s8.size(),3);
BOOST_CHECK_EQUAL ( s8[0], 6);
BOOST_CHECK_EQUAL ( s8[1], 3);
BOOST_CHECK_EQUAL ( s8[2], 1);
}
BOOST_AUTO_TEST_SUITE_END()