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
11 #define BOOST_TEST_MODULE TestUniformIntDistribution
12 #include <boost/test/unit_test.hpp>
13
14 #include <boost/compute/system.hpp>
15 #include <boost/compute/command_queue.hpp>
16 #include <boost/compute/algorithm/count_if.hpp>
17 #include <boost/compute/container/vector.hpp>
18 #include <boost/compute/random/default_random_engine.hpp>
19 #include <boost/compute/random/uniform_int_distribution.hpp>
20 #include <boost/compute/lambda.hpp>
21
22 #include "context_setup.hpp"
23
24 namespace compute=boost::compute;
25
BOOST_AUTO_TEST_CASE(uniform_int_distribution_doctest)26 BOOST_AUTO_TEST_CASE(uniform_int_distribution_doctest)
27 {
28 using boost::compute::uint_;
29 using boost::compute::lambda::_1;
30
31 boost::compute::vector<uint_> vec(128, context);
32
33 //! [generate]
34 // initialize the default random engine
35 boost::compute::default_random_engine engine(queue);
36
37 // setup the uniform distribution to produce integers 0 and 1
38 boost::compute::uniform_int_distribution<uint_> distribution(0, 1);
39
40 // generate the random values and store them to 'vec'
41 distribution.generate(vec.begin(), vec.end(), engine, queue);
42 //! [generate]
43
44 BOOST_CHECK_EQUAL(
45 boost::compute::count_if(
46 vec.begin(), vec.end(), _1 > 1, queue
47 ),
48 size_t(0)
49 );
50 }
51
BOOST_AUTO_TEST_CASE(issue159)52 BOOST_AUTO_TEST_CASE(issue159) {
53 using boost::compute::lambda::_1;
54
55 boost::compute::vector<int> input(10, context);
56
57 // generate random numbers between 1 and 10
58 compute::default_random_engine rng(queue);
59 compute::uniform_int_distribution<int> d(1, 10);
60 d.generate(input.begin(), input.end(), rng, queue);
61
62 BOOST_CHECK_EQUAL(
63 boost::compute::count_if(
64 input.begin(), input.end(), _1 > 10, queue
65 ),
66 size_t(0)
67 );
68
69 BOOST_CHECK_EQUAL(
70 boost::compute::count_if(
71 input.begin(), input.end(), _1 < 1, queue
72 ),
73 size_t(0)
74 );
75 }
76
77 BOOST_AUTO_TEST_SUITE_END()
78