test_set_intersection.cpp 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. //---------------------------------------------------------------------------//
  2. // Copyright (c) 2014 Roshan <thisisroshansmail@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 TestSetIntersection
  11. #include <boost/test/unit_test.hpp>
  12. #include <boost/compute/command_queue.hpp>
  13. #include <boost/compute/algorithm/set_intersection.hpp>
  14. #include <boost/compute/container/vector.hpp>
  15. #include <boost/compute/types/fundamental.hpp>
  16. #include "check_macros.hpp"
  17. #include "context_setup.hpp"
  18. namespace bc = boost::compute;
  19. BOOST_AUTO_TEST_CASE(set_intersection_int)
  20. {
  21. int dataset1[] = {1, 1, 2, 2, 2, 2, 3, 3, 4, 5, 6, 10};
  22. bc::vector<bc::int_> set1(dataset1, dataset1 + 12, queue);
  23. int dataset2[] = {0, 2, 2, 4, 5, 6, 8, 8, 9, 9, 9, 13};
  24. bc::vector<bc::int_> set2(dataset2, dataset2 + 12, queue);
  25. bc::vector<bc::uint_>result(10, queue.get_context());
  26. bc::vector<bc::uint_>::iterator iter =
  27. bc::set_intersection(set1.begin(), set1.begin() + 12,
  28. set2.begin(), set2.begin() + 12,
  29. result.begin(), queue);
  30. CHECK_RANGE_EQUAL(int, 5, result, (2, 2, 4, 5, 6));
  31. BOOST_VERIFY(iter == result.begin()+5);
  32. }
  33. BOOST_AUTO_TEST_CASE(set_intersection_string)
  34. {
  35. char string1[] = "abcccdddeeff";
  36. bc::vector<bc::char_> set1(string1, string1 + 12, queue);
  37. char string2[] = "bccdfgh";
  38. bc::vector<bc::char_> set2(string2, string2 + 7, queue);
  39. bc::vector<bc::char_>result(5, queue.get_context());
  40. bc::vector<bc::char_>::iterator iter =
  41. bc::set_intersection(set1.begin(), set1.begin() + 12,
  42. set2.begin(), set2.begin() + 7,
  43. result.begin(), queue);
  44. CHECK_RANGE_EQUAL(char, 5, result, ('b', 'c', 'c', 'd', 'f'));
  45. BOOST_VERIFY(iter == result.begin()+5);
  46. }
  47. BOOST_AUTO_TEST_SUITE_END()