test_equal_range.cpp 3.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  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 TestEqualRange
  11. #include <boost/test/unit_test.hpp>
  12. #include <utility>
  13. #include <iterator>
  14. #include <boost/compute/command_queue.hpp>
  15. #include <boost/compute/algorithm/equal_range.hpp>
  16. #include <boost/compute/container/vector.hpp>
  17. #include "context_setup.hpp"
  18. BOOST_AUTO_TEST_CASE(equal_range_int)
  19. {
  20. int data[] = { 1, 2, 2, 2, 3, 3, 4, 5 };
  21. boost::compute::vector<int> vector(data, data + 8, queue);
  22. typedef boost::compute::vector<int>::iterator iterator;
  23. std::pair<iterator, iterator> range0 =
  24. boost::compute::equal_range(vector.begin(), vector.end(), int(0), queue);
  25. BOOST_CHECK(range0.first == vector.begin());
  26. BOOST_CHECK(range0.second == vector.begin());
  27. BOOST_CHECK_EQUAL(std::distance(range0.first, range0.second), ptrdiff_t(0));
  28. std::pair<iterator, iterator> range1 =
  29. boost::compute::equal_range(vector.begin(), vector.end(), int(1), queue);
  30. BOOST_CHECK(range1.first == vector.begin());
  31. BOOST_CHECK(range1.second == vector.begin() + 1);
  32. BOOST_CHECK_EQUAL(std::distance(range1.first, range1.second), ptrdiff_t(1));
  33. std::pair<iterator, iterator> range2 =
  34. boost::compute::equal_range(vector.begin(), vector.end(), int(2), queue);
  35. BOOST_CHECK(range2.first == vector.begin() + 1);
  36. BOOST_CHECK(range2.second == vector.begin() + 4);
  37. BOOST_CHECK_EQUAL(std::distance(range2.first, range2.second), ptrdiff_t(3));
  38. std::pair<iterator, iterator> range3 =
  39. boost::compute::equal_range(vector.begin(), vector.end(), int(3), queue);
  40. BOOST_CHECK(range3.first == vector.begin() + 4);
  41. BOOST_CHECK(range3.second == vector.begin() + 6);
  42. BOOST_CHECK_EQUAL(std::distance(range3.first, range3.second), ptrdiff_t(2));
  43. std::pair<iterator, iterator> range4 =
  44. boost::compute::equal_range(vector.begin(), vector.end(), int(4), queue);
  45. BOOST_CHECK(range4.first == vector.begin() + 6);
  46. BOOST_CHECK(range4.second == vector.begin() + 7);
  47. BOOST_CHECK_EQUAL(std::distance(range4.first, range4.second), ptrdiff_t(1));
  48. std::pair<iterator, iterator> range5 =
  49. boost::compute::equal_range(vector.begin(), vector.end(), int(5), queue);
  50. BOOST_CHECK(range5.first == vector.begin() + 7);
  51. BOOST_CHECK(range5.second == vector.end());
  52. BOOST_CHECK_EQUAL(std::distance(range5.first, range5.second), ptrdiff_t(1));
  53. std::pair<iterator, iterator> range6 =
  54. boost::compute::equal_range(vector.begin(), vector.end(), int(6), queue);
  55. BOOST_CHECK(range6.first == vector.end());
  56. BOOST_CHECK(range6.second == vector.end());
  57. BOOST_CHECK_EQUAL(std::distance(range6.first, range6.second), ptrdiff_t(0));
  58. }
  59. BOOST_AUTO_TEST_SUITE_END()