123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252 |
- /* test_piecewise_linear_distribution.cpp
- *
- * Copyright Steven Watanabe 2011
- * 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)
- *
- * $Id$
- *
- */
- #include <boost/random/piecewise_linear_distribution.hpp>
- #include <boost/random/linear_congruential.hpp>
- #include <boost/assign/list_of.hpp>
- #include <sstream>
- #include <vector>
- #include "concepts.hpp"
- #define BOOST_TEST_MAIN
- #include <boost/test/unit_test.hpp>
- using boost::random::test::RandomNumberDistribution;
- using boost::random::piecewise_linear_distribution;
- BOOST_CONCEPT_ASSERT((RandomNumberDistribution< piecewise_linear_distribution<> >));
- struct gen {
- double operator()(double arg) {
- if(arg < 97) return 100;
- else if(arg < 101) return 3;
- else if(arg < 105) return 1;
- else if(arg < 109) return 2;
- else if(arg < 113) return 1;
- else if(arg < 117) return 5;
- else return 100;
- }
- };
- #define CHECK_SEQUENCE(actual, expected) \
- do { \
- std::vector<double> _actual = (actual); \
- std::vector<double> _expected = (expected); \
- BOOST_CHECK_EQUAL_COLLECTIONS( \
- _actual.begin(), _actual.end(), \
- _expected.begin(), _expected.end()); \
- } while(false)
- using boost::assign::list_of;
- BOOST_AUTO_TEST_CASE(test_constructors) {
- boost::random::piecewise_linear_distribution<> dist;
- CHECK_SEQUENCE(dist.intervals(), list_of(0.0)(1.0));
- CHECK_SEQUENCE(dist.densities(), list_of(1.0)(1.0));
- #ifndef BOOST_NO_CXX11_HDR_INITIALIZER_LIST
- boost::random::piecewise_linear_distribution<> dist_il = {
- { 99, 103, 107, 111, 115 },
- gen()
- };
- CHECK_SEQUENCE(dist_il.intervals(), list_of(99)(103)(107)(111)(115));
- CHECK_SEQUENCE(dist_il.densities(),
- list_of(.09375)(.03125)(0.0625)(.03125)(.15625));
- boost::random::piecewise_linear_distribution<> dist_il2 = {
- { 99 },
- gen()
- };
- CHECK_SEQUENCE(dist_il2.intervals(), list_of(0.0)(1.0));
- CHECK_SEQUENCE(dist_il2.densities(), list_of(1.0)(1.0));
- #endif
- std::vector<double> intervals = boost::assign::list_of(0)(1)(2)(3)(5);
- std::vector<double> weights = boost::assign::list_of(3)(1)(2)(1)(2);
- std::vector<double> intervals2 = boost::assign::list_of(99);
- std::vector<double> weights2 = boost::assign::list_of(2);
- boost::random::piecewise_linear_distribution<> dist_r(intervals, weights);
- CHECK_SEQUENCE(dist_r.intervals(), list_of(0)(1)(2)(3)(5));
- CHECK_SEQUENCE(dist_r.densities(), list_of(.375)(.125)(.25)(.125)(.25));
- boost::random::piecewise_linear_distribution<>
- dist_r2(intervals2, weights2);
- CHECK_SEQUENCE(dist_r2.intervals(), list_of(0.0)(1.0));
- CHECK_SEQUENCE(dist_r2.densities(), list_of(1.0)(1.0));
-
- boost::random::piecewise_linear_distribution<> dist_it(
- intervals.begin(), intervals.end(), weights.begin());
- CHECK_SEQUENCE(dist_it.intervals(), list_of(0)(1)(2)(3)(5));
- CHECK_SEQUENCE(dist_it.densities(), list_of(.375)(.125)(.25)(.125)(.25));
-
- boost::random::piecewise_linear_distribution<> dist_it2(
- intervals2.begin(), intervals2.end(), weights2.begin());
- CHECK_SEQUENCE(dist_it2.intervals(), list_of(0.0)(1.0));
- CHECK_SEQUENCE(dist_it2.densities(), list_of(1.0)(1.0));
-
- boost::random::piecewise_linear_distribution<> dist_fun(4, 99,115, gen());
- CHECK_SEQUENCE(dist_fun.intervals(), list_of(99)(103)(107)(111)(115));
- CHECK_SEQUENCE(dist_fun.densities(),
- list_of(.09375)(.03125)(0.0625)(.03125)(.15625));
-
- boost::random::piecewise_linear_distribution<>
- dist_fun2(1, 99, 115, gen());
- CHECK_SEQUENCE(dist_fun2.intervals(), list_of(99)(115));
- CHECK_SEQUENCE(dist_fun2.densities(), list_of(0.046875)(0.078125));
- boost::random::piecewise_linear_distribution<> copy(dist);
- BOOST_CHECK_EQUAL(dist, copy);
- boost::random::piecewise_linear_distribution<> copy_r(dist_r);
- BOOST_CHECK_EQUAL(dist_r, copy_r);
- boost::random::piecewise_linear_distribution<> notpow2(3, 99, 111, gen());
- BOOST_REQUIRE_EQUAL(notpow2.densities().size(), 4u);
- BOOST_CHECK_CLOSE_FRACTION(notpow2.densities()[0], 0.15, 1e-12);
- BOOST_CHECK_CLOSE_FRACTION(notpow2.densities()[1], 0.05, 1e-12);
- BOOST_CHECK_CLOSE_FRACTION(notpow2.densities()[2], 0.1, 1e-12);
- BOOST_CHECK_CLOSE_FRACTION(notpow2.densities()[3], 0.05, 1e-12);
- boost::random::piecewise_linear_distribution<> copy_notpow2(notpow2);
- BOOST_CHECK_EQUAL(notpow2, copy_notpow2);
- }
- BOOST_AUTO_TEST_CASE(test_param) {
- std::vector<double> intervals = boost::assign::list_of(0)(1)(2)(3)(5);
- std::vector<double> weights = boost::assign::list_of(3)(1)(2)(1)(2);
- std::vector<double> intervals2 = boost::assign::list_of(99);
- std::vector<double> weights2 = boost::assign::list_of(2);
- boost::random::piecewise_linear_distribution<> dist(intervals, weights);
- boost::random::piecewise_linear_distribution<>::param_type
- param = dist.param();
- CHECK_SEQUENCE(param.intervals(), list_of(0)(1)(2)(3)(5));
- CHECK_SEQUENCE(param.densities(), list_of(.375)(.125)(.25)(.125)(.25));
- boost::random::piecewise_linear_distribution<> copy1(param);
- BOOST_CHECK_EQUAL(dist, copy1);
- boost::random::piecewise_linear_distribution<> copy2;
- copy2.param(param);
- BOOST_CHECK_EQUAL(dist, copy2);
- boost::random::piecewise_linear_distribution<>::param_type
- param_copy = param;
- BOOST_CHECK_EQUAL(param, param_copy);
- BOOST_CHECK(param == param_copy);
- BOOST_CHECK(!(param != param_copy));
- boost::random::piecewise_linear_distribution<>::param_type param_default;
- CHECK_SEQUENCE(param_default.intervals(), list_of(0.0)(1.0));
- CHECK_SEQUENCE(param_default.densities(), list_of(1.0)(1.0));
- BOOST_CHECK(param != param_default);
- BOOST_CHECK(!(param == param_default));
-
- #ifndef BOOST_NO_CXX11_HDR_INITIALIZER_LIST
- boost::random::piecewise_linear_distribution<>::param_type parm_il = {
- { 99, 103, 107, 111, 115 },
- gen()
- };
- CHECK_SEQUENCE(parm_il.intervals(), list_of(99)(103)(107)(111)(115));
- CHECK_SEQUENCE(parm_il.densities(),
- list_of(.09375)(.03125)(0.0625)(.03125)(.15625));
- boost::random::piecewise_linear_distribution<>::param_type parm_il2 = {
- { 99 },
- gen()
- };
- CHECK_SEQUENCE(parm_il2.intervals(), list_of(0.0)(1.0));
- CHECK_SEQUENCE(parm_il2.densities(), list_of(1.0)(1.0));
- #endif
- boost::random::piecewise_linear_distribution<>::param_type
- parm_r(intervals, weights);
- CHECK_SEQUENCE(parm_r.intervals(), list_of(0)(1)(2)(3)(5));
- CHECK_SEQUENCE(parm_r.densities(), list_of(.375)(.125)(.25)(.125)(.25));
- boost::random::piecewise_linear_distribution<>::param_type
- parm_r2(intervals2, weights2);
- CHECK_SEQUENCE(parm_r2.intervals(), list_of(0.0)(1.0));
- CHECK_SEQUENCE(parm_r2.densities(), list_of(1.0)(1.0));
-
- boost::random::piecewise_linear_distribution<>::param_type
- parm_it(intervals.begin(), intervals.end(), weights.begin());
- CHECK_SEQUENCE(parm_it.intervals(), list_of(0)(1)(2)(3)(5));
- CHECK_SEQUENCE(parm_it.densities(), list_of(.375)(.125)(.25)(.125)(.25));
-
- boost::random::piecewise_linear_distribution<>::param_type
- parm_it2(intervals2.begin(), intervals2.end(), weights2.begin());
- CHECK_SEQUENCE(parm_it2.intervals(), list_of(0.0)(1.0));
- CHECK_SEQUENCE(parm_it2.densities(), list_of(1.0)(1.0));
-
- boost::random::piecewise_linear_distribution<>::param_type
- parm_fun(4, 99, 115, gen());
- CHECK_SEQUENCE(parm_fun.intervals(), list_of(99)(103)(107)(111)(115));
- CHECK_SEQUENCE(parm_fun.densities(),
- list_of(.09375)(.03125)(0.0625)(.03125)(.15625));
-
- boost::random::piecewise_linear_distribution<>::param_type
- parm_fun2(1, 99, 115, gen());
- CHECK_SEQUENCE(parm_fun2.intervals(), list_of(99)(115));
- CHECK_SEQUENCE(parm_fun2.densities(), list_of(0.046875)(0.078125));
- }
- BOOST_AUTO_TEST_CASE(test_min_max) {
- std::vector<double> intervals = boost::assign::list_of(0)(1)(2)(3)(5);
- std::vector<double> weights = boost::assign::list_of(3)(1)(2)(1)(2);
- boost::random::piecewise_linear_distribution<> dist;
- BOOST_CHECK_EQUAL((dist.min)(), 0.0);
- BOOST_CHECK_EQUAL((dist.max)(), 1.0);
- boost::random::piecewise_linear_distribution<> dist_r(intervals, weights);
- BOOST_CHECK_EQUAL((dist_r.min)(), 0.0);
- BOOST_CHECK_EQUAL((dist_r.max)(), 5.0);
- }
- BOOST_AUTO_TEST_CASE(test_comparison) {
- std::vector<double> intervals = boost::assign::list_of(0)(1)(2)(3)(5);
- std::vector<double> weights = boost::assign::list_of(3)(1)(2)(1)(2);
- boost::random::piecewise_linear_distribution<> dist;
- boost::random::piecewise_linear_distribution<> dist_copy(dist);
- boost::random::piecewise_linear_distribution<> dist_r(intervals, weights);
- boost::random::piecewise_linear_distribution<> dist_r_copy(dist_r);
- BOOST_CHECK(dist == dist_copy);
- BOOST_CHECK(!(dist != dist_copy));
- BOOST_CHECK(dist_r == dist_r_copy);
- BOOST_CHECK(!(dist_r != dist_r_copy));
- BOOST_CHECK(dist != dist_r);
- BOOST_CHECK(!(dist == dist_r));
- }
- BOOST_AUTO_TEST_CASE(test_streaming) {
- std::vector<double> intervals = boost::assign::list_of(0)(1)(2)(3)(5);
- std::vector<double> weights = boost::assign::list_of(3)(1)(2)(1)(2);
- boost::random::piecewise_linear_distribution<> dist(intervals, weights);
- std::stringstream stream;
- stream << dist;
- boost::random::piecewise_linear_distribution<> restored_dist;
- stream >> restored_dist;
- BOOST_CHECK_EQUAL(dist, restored_dist);
- }
- BOOST_AUTO_TEST_CASE(test_generation) {
- std::vector<double> intervals = boost::assign::list_of(1)(2);
- std::vector<double> weights = boost::assign::list_of(1)(1);
- boost::minstd_rand0 gen;
- boost::random::piecewise_linear_distribution<> dist;
- boost::random::piecewise_linear_distribution<> dist_r(intervals, weights);
- for(int i = 0; i < 10; ++i) {
- double value = dist(gen);
- BOOST_CHECK_GE(value, 0.0);
- BOOST_CHECK_LT(value, 1.0);
- double value_r = dist_r(gen);
- BOOST_CHECK_GE(value_r, 1.0);
- BOOST_CHECK_LT(value_r, 2.0);
- double value_param = dist_r(gen, dist.param());
- BOOST_CHECK_GE(value_param, 0.0);
- BOOST_CHECK_LT(value_param, 1.0);
- double value_r_param = dist(gen, dist_r.param());
- BOOST_CHECK_GE(value_r_param, 1.0);
- BOOST_CHECK_LT(value_r_param, 2.0);
- }
- }
|