ceph/src/boost/libs/compute/test/test_extents.cpp

   1 //---------------------------------------------------------------------------//
   2 // Copyright (c) 2013-2014 Kyle Lutz <kyle.r.lutz@gmail.com>
   3 //
   4 // Distributed under the Boost Software License, Version 1.0
   5 // See accompanying file LICENSE_1_0.txt or copy at
   6 // http://www.boost.org/LICENSE_1_0.txt
   7 //
   8 // See http://boostorg.github.com/compute for more information.
   9 //---------------------------------------------------------------------------//
  10
  11 #define BOOST_TEST_MODULE TestExtents
  12 #include <boost/test/unit_test.hpp>
  13
  14 #include <algorithm>
  15 #include <vector>
  16
  17 #include <boost/compute/utility/dim.hpp>
  18 #include <boost/compute/utility/extents.hpp>
  19
  20 #include "context_setup.hpp"
  21
  22 namespace compute = boost::compute;
  23
  24 BOOST_AUTO_TEST_CASE(initialize)
  25 {
  26     compute::extents<1> one(1);
  27     BOOST_CHECK_EQUAL(one[0], size_t(1));
  28
  29     compute::extents<3> xyz = compute::dim(1, 2, 3);
  30     BOOST_CHECK_EQUAL(xyz[0], size_t(1));
  31     BOOST_CHECK_EQUAL(xyz[1], size_t(2));
  32     BOOST_CHECK_EQUAL(xyz[2], size_t(3));
  33 }
  34
  35 BOOST_AUTO_TEST_CASE(size)
  36 {
  37     BOOST_CHECK_EQUAL(compute::extents<1>().size(), size_t(1));
  38     BOOST_CHECK_EQUAL(compute::extents<2>().size(), size_t(2));
  39     BOOST_CHECK_EQUAL(compute::extents<3>().size(), size_t(3));
  40 }
  41
  42 BOOST_AUTO_TEST_CASE(subscript_operator)
  43 {
  44     compute::extents<3> xyz;
  45     BOOST_CHECK_EQUAL(xyz[0], size_t(0));
  46     BOOST_CHECK_EQUAL(xyz[1], size_t(0));
  47     BOOST_CHECK_EQUAL(xyz[2], size_t(0));
  48
  49     xyz[0] = size_t(10);
  50     xyz[1] = size_t(20);
  51     xyz[2] = size_t(30);
  52     BOOST_CHECK_EQUAL(xyz[0], size_t(10));
  53     BOOST_CHECK_EQUAL(xyz[1], size_t(20));
  54     BOOST_CHECK_EQUAL(xyz[2], size_t(30));
  55 }
  56
  57 BOOST_AUTO_TEST_CASE(data)
  58 {
  59     compute::extents<3> xyz = compute::dim(5, 6, 7);
  60     BOOST_CHECK_EQUAL(xyz.data()[0], size_t(5));
  61     BOOST_CHECK_EQUAL(xyz.data()[1], size_t(6));
  62     BOOST_CHECK_EQUAL(xyz.data()[2], size_t(7));
  63 }
  64
  65 BOOST_AUTO_TEST_CASE(linear)
  66 {
  67     compute::extents<2> uv = compute::dim(16, 16);
  68     BOOST_CHECK_EQUAL(uv.linear(), size_t(256));
  69 }
  70
  71 BOOST_AUTO_TEST_CASE(equality_operator)
  72 {
  73     BOOST_CHECK(compute::dim(10, 20) == compute::dim(10, 20));
  74     BOOST_CHECK(compute::dim(20, 10) != compute::dim(10, 20));
  75 }
  76
  77 BOOST_AUTO_TEST_CASE(empty_dim)
  78 {
  79     BOOST_CHECK(compute::dim<0>() == compute::dim());
  80     BOOST_CHECK(compute::dim<1>() == compute::dim(0));
  81     BOOST_CHECK(compute::dim<2>() == compute::dim(0, 0));
  82     BOOST_CHECK(compute::dim<3>() == compute::dim(0, 0, 0));
  83 }
  84
  85 BOOST_AUTO_TEST_CASE(copy_to_vector)
  86 {
  87     compute::extents<3> exts = compute::dim(4, 5, 6);
  88
  89     std::vector<size_t> vec(3);
  90     std::copy(exts.begin(), exts.end(), vec.begin());
  91     BOOST_CHECK_EQUAL(vec[0], size_t(4));
  92     BOOST_CHECK_EQUAL(vec[1], size_t(5));
  93     BOOST_CHECK_EQUAL(vec[2], size_t(6));
  94 }
  95
  96 BOOST_AUTO_TEST_SUITE_END()