compute/test/test_mersenne_twister_engine.cpp
2015-12-28 00:13:53 +09:00

121 lines
2.9 KiB
C++

//---------------------------------------------------------------------------//
// Copyright (c) 2013 Kyle Lutz <kyle.r.lutz@gmail.com>
//
// Distributed under the Boost Software License, Version 1.0
// See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt
//
// See http://boostorg.github.com/compute for more information.
//---------------------------------------------------------------------------//
#define BOOST_TEST_MODULE TestMersenneTwisterEngine
#include <boost/test/unit_test.hpp>
#include <boost/compute/random/mersenne_twister_engine.hpp>
#include <boost/compute/container/vector.hpp>
#include "check_macros.hpp"
#include "context_setup.hpp"
BOOST_AUTO_TEST_CASE(generate_uint)
{
using boost::compute::uint_;
boost::compute::mt19937 rng(queue);
boost::compute::vector<uint_> vector(10, context);
rng.generate(vector.begin(), vector.end(), queue);
CHECK_RANGE_EQUAL(
uint_, 10, vector,
(uint_(3499211612),
uint_(581869302),
uint_(3890346734),
uint_(3586334585),
uint_(545404204),
uint_(4161255391),
uint_(3922919429),
uint_(949333985),
uint_(2715962298),
uint_(1323567403))
);
}
BOOST_AUTO_TEST_CASE(discard_uint)
{
using boost::compute::uint_;
boost::compute::mt19937 rng(queue);
boost::compute::vector<uint_> vector(5, context);
rng.discard(5, queue);
rng.generate(vector.begin(), vector.end(), queue);
CHECK_RANGE_EQUAL(
uint_, 5, vector,
(uint_(4161255391),
uint_(3922919429),
uint_(949333985),
uint_(2715962298),
uint_(1323567403))
);
}
BOOST_AUTO_TEST_CASE(copy_ctor)
{
using boost::compute::uint_;
boost::compute::mt19937 rng(queue);
boost::compute::mt19937 rng_copy(rng);
boost::compute::vector<uint_> vector(10, context);
rng_copy.generate(vector.begin(), vector.end(), queue);
CHECK_RANGE_EQUAL(
uint_, 10, vector,
(uint_(3499211612),
uint_(581869302),
uint_(3890346734),
uint_(3586334585),
uint_(545404204),
uint_(4161255391),
uint_(3922919429),
uint_(949333985),
uint_(2715962298),
uint_(1323567403))
);
}
BOOST_AUTO_TEST_CASE(assign_op)
{
using boost::compute::uint_;
boost::compute::mt19937 rng(queue);
boost::compute::mt19937 rng_copy(queue);
boost::compute::vector<uint_> vector(10, context);
rng_copy.discard(5, queue);
rng_copy = rng;
rng_copy.generate(vector.begin(), vector.end(), queue);
CHECK_RANGE_EQUAL(
uint_, 10, vector,
(uint_(3499211612),
uint_(581869302),
uint_(3890346734),
uint_(3586334585),
uint_(545404204),
uint_(4161255391),
uint_(3922919429),
uint_(949333985),
uint_(2715962298),
uint_(1323567403))
);
}
BOOST_AUTO_TEST_SUITE_END()