test_pair.cpp 6.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194
  1. //---------------------------------------------------------------------------//
  2. // Copyright (c) 2013 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. #define BOOST_TEST_MODULE TestPair
  11. #include <boost/test/unit_test.hpp>
  12. #include <iostream>
  13. #include <boost/compute/algorithm/copy.hpp>
  14. #include <boost/compute/algorithm/fill.hpp>
  15. #include <boost/compute/algorithm/find.hpp>
  16. #include <boost/compute/algorithm/transform.hpp>
  17. #include <boost/compute/container/vector.hpp>
  18. #include <boost/compute/functional/get.hpp>
  19. #include <boost/compute/functional/field.hpp>
  20. #include <boost/compute/types/pair.hpp>
  21. #include "quirks.hpp"
  22. #include "check_macros.hpp"
  23. #include "context_setup.hpp"
  24. BOOST_AUTO_TEST_CASE(vector_pair_int_float)
  25. {
  26. boost::compute::vector<std::pair<int, float> > vector(context);
  27. vector.push_back(std::make_pair(1, 1.1f), queue);
  28. vector.push_back(std::make_pair(2, 2.2f), queue);
  29. vector.push_back(std::make_pair(3, 3.3f), queue);
  30. queue.finish();
  31. BOOST_CHECK_EQUAL(vector.size(), size_t(3));
  32. BOOST_CHECK(vector[0] == std::make_pair(1, 1.1f));
  33. BOOST_CHECK(vector[1] == std::make_pair(2, 2.2f));
  34. BOOST_CHECK(vector[2] == std::make_pair(3, 3.3f));
  35. }
  36. BOOST_AUTO_TEST_CASE(copy_pair_vector)
  37. {
  38. boost::compute::vector<std::pair<int, float> > input(context);
  39. input.push_back(std::make_pair(1, 2.0f), queue);
  40. input.push_back(std::make_pair(3, 4.0f), queue);
  41. input.push_back(std::make_pair(5, 6.0f), queue);
  42. input.push_back(std::make_pair(7, 8.0f), queue);
  43. queue.finish();
  44. BOOST_CHECK_EQUAL(input.size(), size_t(4));
  45. boost::compute::vector<std::pair<int, float> > output(4, context);
  46. boost::compute::copy(input.begin(), input.end(), output.begin(), queue);
  47. queue.finish();
  48. BOOST_CHECK(output[0] == std::make_pair(1, 2.0f));
  49. BOOST_CHECK(output[1] == std::make_pair(3, 4.0f));
  50. BOOST_CHECK(output[2] == std::make_pair(5, 6.0f));
  51. BOOST_CHECK(output[3] == std::make_pair(7, 8.0f));
  52. }
  53. BOOST_AUTO_TEST_CASE(fill_pair_vector)
  54. {
  55. if(bug_in_struct_assignment(device)){
  56. std::cerr << "skipping fill_pair_vector test" << std::endl;
  57. return;
  58. }
  59. boost::compute::vector<std::pair<int, float> > vector(5, context);
  60. boost::compute::fill(vector.begin(), vector.end(), std::make_pair(4, 2.0f), queue);
  61. queue.finish();
  62. BOOST_CHECK(vector[0] == std::make_pair(4, 2.0f));
  63. BOOST_CHECK(vector[1] == std::make_pair(4, 2.0f));
  64. BOOST_CHECK(vector[2] == std::make_pair(4, 2.0f));
  65. BOOST_CHECK(vector[3] == std::make_pair(4, 2.0f));
  66. BOOST_CHECK(vector[4] == std::make_pair(4, 2.0f));
  67. }
  68. BOOST_AUTO_TEST_CASE(fill_char_pair_vector)
  69. {
  70. if(bug_in_struct_assignment(device)){
  71. std::cerr << "skipping fill_char_pair_vector test" << std::endl;
  72. return;
  73. }
  74. std::pair<char, unsigned char> value('c', static_cast<unsigned char>(127));
  75. boost::compute::vector<std::pair<char, unsigned char> > vector(5, context);
  76. boost::compute::fill(vector.begin(), vector.end(), value, queue);
  77. queue.finish();
  78. BOOST_CHECK(vector[0] == value);
  79. BOOST_CHECK(vector[1] == value);
  80. BOOST_CHECK(vector[2] == value);
  81. BOOST_CHECK(vector[3] == value);
  82. BOOST_CHECK(vector[4] == value);
  83. }
  84. BOOST_AUTO_TEST_CASE(transform_pair_get)
  85. {
  86. boost::compute::vector<std::pair<int, float> > input(context);
  87. input.push_back(std::make_pair(1, 2.0f), queue);
  88. input.push_back(std::make_pair(3, 4.0f), queue);
  89. input.push_back(std::make_pair(5, 6.0f), queue);
  90. input.push_back(std::make_pair(7, 8.0f), queue);
  91. queue.finish();
  92. boost::compute::vector<int> first_output(4, context);
  93. boost::compute::transform(
  94. input.begin(),
  95. input.end(),
  96. first_output.begin(),
  97. ::boost::compute::get<0>(),
  98. queue
  99. );
  100. CHECK_RANGE_EQUAL(int, 4, first_output, (1, 3, 5, 7));
  101. boost::compute::vector<float> second_output(4, context);
  102. boost::compute::transform(
  103. input.begin(),
  104. input.end(),
  105. second_output.begin(),
  106. ::boost::compute::get<1>(),
  107. queue
  108. );
  109. CHECK_RANGE_EQUAL(float, 4, second_output, (2.0f, 4.0f, 6.0f, 8.0f));
  110. }
  111. BOOST_AUTO_TEST_CASE(transform_pair_field)
  112. {
  113. boost::compute::vector<std::pair<int, float> > input(context);
  114. input.push_back(std::make_pair(1, 2.0f), queue);
  115. input.push_back(std::make_pair(3, 4.0f), queue);
  116. input.push_back(std::make_pair(5, 6.0f), queue);
  117. input.push_back(std::make_pair(7, 8.0f), queue);
  118. boost::compute::vector<int> first_output(4, context);
  119. boost::compute::transform(
  120. input.begin(),
  121. input.end(),
  122. first_output.begin(),
  123. boost::compute::field<int>("first"),
  124. queue
  125. );
  126. CHECK_RANGE_EQUAL(int, 4, first_output, (1, 3, 5, 7));
  127. boost::compute::vector<float> second_output(4, context);
  128. boost::compute::transform(
  129. input.begin(),
  130. input.end(),
  131. second_output.begin(),
  132. boost::compute::field<float>("second"),
  133. queue
  134. );
  135. CHECK_RANGE_EQUAL(float, 4, second_output, (2.0f, 4.0f, 6.0f, 8.0f));
  136. }
  137. BOOST_AUTO_TEST_CASE(find_vector_pair)
  138. {
  139. boost::compute::vector<std::pair<int, float> > vector(context);
  140. vector.push_back(std::make_pair(1, 1.1f), queue);
  141. vector.push_back(std::make_pair(2, 2.2f), queue);
  142. vector.push_back(std::make_pair(3, 3.3f), queue);
  143. BOOST_CHECK_EQUAL(vector.size(), size_t(3));
  144. BOOST_CHECK(
  145. boost::compute::find(
  146. boost::compute::make_transform_iterator(
  147. vector.begin(),
  148. boost::compute::get<0>()
  149. ),
  150. boost::compute::make_transform_iterator(
  151. vector.end(),
  152. boost::compute::get<0>()
  153. ),
  154. int(2),
  155. queue
  156. ).base() == vector.begin() + 1
  157. );
  158. BOOST_CHECK(
  159. boost::compute::find(
  160. boost::compute::make_transform_iterator(
  161. vector.begin(),
  162. boost::compute::get<1>()
  163. ),
  164. boost::compute::make_transform_iterator(
  165. vector.end(),
  166. boost::compute::get<1>()
  167. ),
  168. float(3.3f),
  169. queue
  170. ).base() == vector.begin() + 2
  171. );
  172. }
  173. BOOST_AUTO_TEST_SUITE_END()