mirror of
				https://github.com/saitohirga/WSJT-X.git
				synced 2025-10-31 13:10:19 -04:00 
			
		
		
		
	
		
			
				
	
	
		
			803 lines
		
	
	
		
			31 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			803 lines
		
	
	
		
			31 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| // test_geometric.cpp
 | |
| 
 | |
| // Copyright Paul A. Bristow 2010.
 | |
| // Copyright John Maddock 2010.
 | |
| 
 | |
| // Use, modification and distribution are subject to 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)
 | |
| 
 | |
| // Tests for Geometric Distribution.
 | |
| 
 | |
| // Note that these defines must be placed BEFORE #includes.
 | |
| #define BOOST_MATH_OVERFLOW_ERROR_POLICY ignore_error
 | |
| // because several tests overflow & underflow by design.
 | |
| #define BOOST_MATH_DISCRETE_QUANTILE_POLICY real
 | |
| 
 | |
| #ifdef _MSC_VER
 | |
| #  pragma warning(disable: 4127) // conditional expression is constant.
 | |
| #endif
 | |
| 
 | |
| #if !defined(TEST_FLOAT) && !defined(TEST_DOUBLE) && !defined(TEST_LDOUBLE) && !defined(TEST_REAL_CONCEPT)
 | |
| #  define TEST_FLOAT
 | |
| #  define TEST_DOUBLE
 | |
| #  define TEST_LDOUBLE
 | |
| #  define TEST_REAL_CONCEPT
 | |
| #endif
 | |
| 
 | |
| #include <boost/math/tools/test.hpp>
 | |
| #include <boost/math/concepts/real_concept.hpp> // for real_concept
 | |
| using ::boost::math::concepts::real_concept;
 | |
| 
 | |
| #include <boost/math/distributions/geometric.hpp> // for geometric_distribution
 | |
| using boost::math::geometric_distribution;
 | |
| using boost::math::geometric; // using typedef for geometric_distribution<double>
 | |
| 
 | |
| #include <boost/math/distributions/negative_binomial.hpp> // for some comparisons.
 | |
| 
 | |
| #define BOOST_TEST_MAIN
 | |
| #include <boost/test/unit_test.hpp> // for test_main
 | |
| #include <boost/test/floating_point_comparison.hpp> // for BOOST_CHECK_CLOSE_FRACTION
 | |
| #include "test_out_of_range.hpp"
 | |
| 
 | |
| #include <iostream>
 | |
| using std::cout;
 | |
| using std::endl;
 | |
| using std::setprecision;
 | |
| using std::showpoint;
 | |
| #include <limits>
 | |
| using std::numeric_limits;
 | |
| 
 | |
| template <class RealType>
 | |
| void test_spot( // Test a single spot value against 'known good' values.
 | |
|                RealType k,    // Number of failures.
 | |
|                RealType p,    // Probability of success_fraction.
 | |
|                RealType P,    // CDF probability.
 | |
|                RealType Q,    // Complement of CDF.
 | |
|                RealType tol)  // Test tolerance.
 | |
| {
 | |
|    boost::math::geometric_distribution<RealType> g(p);
 | |
|    BOOST_CHECK_EQUAL(p, g.success_fraction());
 | |
|    BOOST_CHECK_CLOSE_FRACTION(cdf(g, k), P, tol);
 | |
| 
 | |
|   if((P < 0.99) && (Q < 0.99))
 | |
|   {
 | |
|     // We can only check this if P is not too close to 1,
 | |
|     // so that we can guarantee that Q is free of error:
 | |
|     //
 | |
|     BOOST_CHECK_CLOSE_FRACTION(
 | |
|       cdf(complement(g, k)), Q, tol);
 | |
|     if(k != 0)
 | |
|     {
 | |
|       BOOST_CHECK_CLOSE_FRACTION(
 | |
|         quantile(g, P), k, tol);
 | |
|     }
 | |
|     else
 | |
|     {
 | |
|       // Just check quantile is very small:
 | |
|       if((std::numeric_limits<RealType>::max_exponent <= std::numeric_limits<double>::max_exponent)
 | |
|         && (boost::is_floating_point<RealType>::value))
 | |
|       {
 | |
|         // Limit where this is checked: if exponent range is very large we may
 | |
|         // run out of iterations in our root finding algorithm.
 | |
|         BOOST_CHECK(quantile(g, P) < boost::math::tools::epsilon<RealType>() * 10);
 | |
|       }
 | |
|     }
 | |
|     if(k != 0)
 | |
|     {
 | |
|       BOOST_CHECK_CLOSE_FRACTION(
 | |
|         quantile(complement(g, Q)), k, tol);
 | |
|     }
 | |
|     else
 | |
|     {
 | |
|       // Just check quantile is very small:
 | |
|       if((std::numeric_limits<RealType>::max_exponent <= std::numeric_limits<double>::max_exponent)
 | |
|         && (boost::is_floating_point<RealType>::value))
 | |
|       {
 | |
|         // Limit where this is checked: if exponent range is very large we may
 | |
|         // run out of iterations in our root finding algorithm.
 | |
|         BOOST_CHECK(quantile(complement(g, Q)) < boost::math::tools::epsilon<RealType>() * 10);
 | |
|       }
 | |
|     }
 | |
|   } //   if((P < 0.99) && (Q < 0.99))
 | |
| 
 | |
|     // Parameter estimation test:  estimate success ratio:
 | |
|     BOOST_CHECK_CLOSE_FRACTION(
 | |
|       geometric_distribution<RealType>::find_lower_bound_on_p(
 | |
|       1+k, P),
 | |
|       p, 0.02); // Wide tolerance needed for some tests.
 | |
|    // Note we bump up the sample size here, purely for the sake of the test,
 | |
|     // internally the function has to adjust the sample size so that we get
 | |
|     // the right upper bound, our test undoes this, so we can verify the result.
 | |
|     BOOST_CHECK_CLOSE_FRACTION(
 | |
|       geometric_distribution<RealType>::find_upper_bound_on_p(
 | |
|       1+k+1, Q),
 | |
|       p, 0.02);
 | |
| 
 | |
|     if(Q < P)
 | |
|     {
 | |
|        //
 | |
|        // We check two things here, that the upper and lower bounds
 | |
|        // are the right way around, and that they do actually bracket
 | |
|        // the naive estimate of p = successes / (sample size)
 | |
|        //
 | |
|       BOOST_CHECK(
 | |
|         geometric_distribution<RealType>::find_lower_bound_on_p(
 | |
|         1+k, Q)
 | |
|         <=
 | |
|         geometric_distribution<RealType>::find_upper_bound_on_p(
 | |
|         1+k, Q)
 | |
|         );
 | |
|       BOOST_CHECK(
 | |
|         geometric_distribution<RealType>::find_lower_bound_on_p(
 | |
|         1+k, Q)
 | |
|         <=
 | |
|         1 / (1+k)
 | |
|         );
 | |
|       BOOST_CHECK(
 | |
|         1 / (1+k)
 | |
|         <=
 | |
|         geometric_distribution<RealType>::find_upper_bound_on_p(
 | |
|         1+k, Q)
 | |
|         );
 | |
|     }
 | |
|     else
 | |
|     {
 | |
|        // As above but when P is small.
 | |
|       BOOST_CHECK(
 | |
|         geometric_distribution<RealType>::find_lower_bound_on_p(
 | |
|         1+k, P)
 | |
|         <=
 | |
|         geometric_distribution<RealType>::find_upper_bound_on_p(
 | |
|         1+k, P)
 | |
|         );
 | |
|       BOOST_CHECK(
 | |
|         geometric_distribution<RealType>::find_lower_bound_on_p(
 | |
|         1+k,  P)
 | |
|         <=
 | |
|         1 / (1+k)
 | |
|         );
 | |
|       BOOST_CHECK(
 | |
|         1 / (1+k)
 | |
|         <=
 | |
|         geometric_distribution<RealType>::find_upper_bound_on_p(
 | |
|         1+k, P)
 | |
|         );
 | |
|     }
 | |
| 
 | |
|     // Estimate sample size:
 | |
|     BOOST_CHECK_CLOSE_FRACTION(
 | |
|       geometric_distribution<RealType>::find_minimum_number_of_trials(
 | |
|       k, p, P),
 | |
|       1+k, 0.02); // Can differ 50 to 51 for small p
 | |
|     BOOST_CHECK_CLOSE_FRACTION(
 | |
|       geometric_distribution<RealType>::find_maximum_number_of_trials(
 | |
|          k, p, Q),
 | |
|       1+k, 0.02);
 | |
| 
 | |
| } // test_spot
 | |
| 
 | |
| template <class RealType> // Any floating-point type RealType.
 | |
| void test_spots(RealType)
 | |
| {
 | |
|   // Basic sanity checks.
 | |
|   // Most test data is to double precision (17 decimal digits) only,
 | |
| 
 | |
|   cout << "Floating point Type is " << typeid(RealType).name() << endl;
 | |
| 
 | |
|   // so set tolerance to 1000 eps expressed as a fraction,
 | |
|   // or 1000 eps of type double expressed as a fraction,
 | |
|   // whichever is the larger.
 | |
| 
 | |
|   RealType tolerance = (std::max)
 | |
|     (boost::math::tools::epsilon<RealType>(),
 | |
|     static_cast<RealType>(std::numeric_limits<double>::epsilon()));
 | |
|   tolerance *= 10; // 10 eps
 | |
| 
 | |
|   cout << "Tolerance = " << tolerance << "." << endl;
 | |
| 
 | |
|   RealType tol1eps = boost::math::tools::epsilon<RealType>(); // Very tight, suit exact values.
 | |
|   //RealType tol2eps = boost::math::tools::epsilon<RealType>() * 2; // Tight,  values.
 | |
|   RealType tol5eps = boost::math::tools::epsilon<RealType>() * 5; // Wider 5 epsilon.
 | |
|   cout << "Tolerance 5 eps = " << tol5eps << "." << endl;
 | |
| 
 | |
| 
 | |
|   // Sources of spot test values are mainly R.
 | |
| 
 | |
|   using boost::math::geometric_distribution;
 | |
|   using boost::math::geometric;
 | |
|   using boost::math::cdf;
 | |
|   using boost::math::pdf;
 | |
|   using boost::math::quantile;
 | |
|   using boost::math::complement;
 | |
| 
 | |
|   BOOST_MATH_STD_USING // for std math functions
 | |
| 
 | |
|   // Test geometric using cdf spot values R
 | |
|   // These test quantiles and complements as well.
 | |
| 
 | |
|   test_spot(  //
 | |
|   static_cast<RealType>(2),   // Number of failures, k
 | |
|   static_cast<RealType>(0.5), // Probability of success as fraction, p
 | |
|   static_cast<RealType>(0.875L), // Probability of result (CDF), P
 | |
|   static_cast<RealType>(0.125L),  // complement CCDF Q = 1 - P
 | |
|   tolerance);
 | |
| 
 | |
|   test_spot( //
 | |
|   static_cast<RealType>(0),    // Number of failures, k
 | |
|   static_cast<RealType>(0.25), // Probability of success as fraction, p
 | |
|   static_cast<RealType>(0.25),   // Probability of result (CDF), P
 | |
|   static_cast<RealType>(0.75),   // Q = 1 - P
 | |
|   tolerance);
 | |
| 
 | |
|   test_spot(
 | |
|     // R formatC(pgeom(10,0.25), digits=17) [1] "0.95776486396789551"
 | |
|     // formatC(pgeom(10,0.25, FALSE), digits=17) [1] "0.042235136032104499"
 | |
| 
 | |
|   static_cast<RealType>(10),  // Number of failures, k
 | |
|   static_cast<RealType>(0.25),  // Probability of success, p
 | |
|   static_cast<RealType>(0.95776486396789551L),  // Probability of result (CDF), P
 | |
|   static_cast<RealType>(0.042235136032104499L), // Q = 1 - P
 | |
|   tolerance);
 | |
| 
 | |
|   test_spot(  //
 | |
|   // > R formatC(pgeom(50,0.25, TRUE), digits=17) [1] "0.99999957525875771"
 | |
|   // > R formatC(pgeom(50,0.25, FALSE), digits=17) [1] "4.2474124232020353e-07"
 | |
|   static_cast<RealType>(50),     // Number of failures, k
 | |
|   static_cast<RealType>(0.25),     // Probability of success, p
 | |
|   static_cast<RealType>(0.99999957525875771),  // Probability of result (CDF), P
 | |
|   static_cast<RealType>(4.2474124232020353e-07),   // Q = 1 - P
 | |
|   tolerance);
 | |
|   /*
 | |
|   // This causes failures in find_upper_bound_on_p p is small branch.
 | |
|   test_spot(  // formatC(pgeom(50,0.01, TRUE), digits=17)[1] "0.40104399353383874"
 | |
|     // > formatC(pgeom(50,0.01, FALSE), digits=17) [1] "0.59895600646616121"
 | |
|   static_cast<RealType>(50), // Number of failures, k
 | |
|   static_cast<RealType>(0.01),   // Probability of success, p
 | |
|   static_cast<RealType>(0.40104399353383874),   // Probability of result (CDF), P
 | |
|   static_cast<RealType>(0.59895600646616121),   // Q = 1 - P
 | |
|   tolerance);
 | |
|   */
 | |
| 
 | |
|   test_spot( // > formatC(pgeom(50,0.99, TRUE), digits=17) [1] "                 1"
 | |
|     // formatC(pgeom(50,0.99, FALSE), digits=17) [1] "1.0000000000000364e-102"
 | |
|   static_cast<RealType>(50),     // Number of failures, k
 | |
|   static_cast<RealType>(0.99),    // Probability of success, p
 | |
|   static_cast<RealType>(1), // Probability of result (CDF), P
 | |
|   static_cast<RealType>(1.0000000000000364e-102),   // Q = 1 - P
 | |
|   tolerance);
 | |
| 
 | |
|   test_spot(  // > formatC(pgeom(1,0.99, TRUE), digits=17) [1] "0.99990000000000001"
 | |
|     // > formatC(pgeom(1,0.99, FALSE), digits=17) [1] "0.00010000000000000009"
 | |
|   static_cast<RealType>(1),     // Number of failures, k
 | |
|   static_cast<RealType>(0.99),                    // Probability of success, p
 | |
|   static_cast<RealType>(0.9999),     // Probability of result (CDF), P
 | |
|   static_cast<RealType>(0.0001),   // Q = 1 - P
 | |
|   tolerance);
 | |
| 
 | |
| if(std::numeric_limits<RealType>::is_specialized)
 | |
| { // An extreme value test that is more accurate than using negative binomial.
 | |
|   // Since geometric only uses exp and log functions.
 | |
|   test_spot(  // > formatC(pgeom(10000, 0.001, TRUE), digits=17) [1] "0.99995487182736897"
 | |
| // > formatC(pgeom(10000,0.001, FALSE), digits=17) [1] "4.5128172631071587e-05"
 | |
|   static_cast<RealType>(10000L), // Number of failures, k
 | |
|   static_cast<RealType>(0.001L),                    // Probability of success, p
 | |
|   static_cast<RealType>(0.99995487182736897L),     // Probability of result (CDF), P
 | |
|   static_cast<RealType>(4.5128172631071587e-05L),   // Q = 1 - P
 | |
|   tolerance); //
 | |
|   } // numeric_limit is specialized
 | |
|  // End of single spot tests using RealType
 | |
| 
 | |
|   // Tests on PDF:
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //> formatC(dgeom(0,0.5), digits=17)[1] " 0.5"
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.5)),
 | |
|   static_cast<RealType>(0.0) ),  // Number of failures, k is very small but not integral,
 | |
|   static_cast<RealType>(0.5), // nearly success probability.
 | |
|   tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //> formatC(dgeom(0,0.5), digits=17)[1] "    0.5"
 | |
|     //  R treates geom as a discrete distribution.
 | |
|     // > formatC(dgeom(1.999999,0.5, FALSE), digits=17) [1] "   0"
 | |
|     // Warning message:
 | |
|     // In dgeom(1.999999, 0.5, FALSE) : non-integer x = 1.999999
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.5)),
 | |
|   static_cast<RealType>(0.0001L) ),  // Number of failures, k is very small but not integral,
 | |
|   static_cast<RealType>(0.4999653438420768L), // nearly success probability.
 | |
|   tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( // > formatC(pgeom(0.0001,0.5, TRUE), digits=17)[1] " 0.5"
 | |
|     // > formatC(pgeom(0.0001,0.5, FALSE), digits=17) [1] "               0.5"
 | |
|     //  R treates geom as a discrete distribution.
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.5)),
 | |
|   static_cast<RealType>(0.0001L) ),  // Number of failures, k is very small but not integral,
 | |
|   static_cast<RealType>(0.4999653438420768L), // nearly success probability.
 | |
|   tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( // formatC(dgeom(1,0.01), digits=17)[1] "0.0099000000000000008"
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.01L)),
 | |
|   static_cast<RealType>(1) ),  // Number of failures, k
 | |
|   static_cast<RealType>(0.0099000000000000008), //
 | |
|   tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //> formatC(dgeom(1,0.99), digits=17)[1] "0.0099000000000000043"
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.99L)),
 | |
|   static_cast<RealType>(1) ),  // Number of failures, k
 | |
|   static_cast<RealType>(0.00990000000000000043L), //
 | |
|   tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //> > formatC(dgeom(0,0.99), digits=17)[1] "0.98999999999999999"
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.99L)),
 | |
|   static_cast<RealType>(0) ),  // Number of failures, k
 | |
|   static_cast<RealType>(0.98999999999999999L), //
 | |
|   tolerance);
 | |
| 
 | |
|   // p  near unity.
 | |
|   BOOST_CHECK_CLOSE_FRACTION( // > formatC(dgeom(100,0.99), digits=17)[1] "9.9000000000003448e-201"
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.99L)),
 | |
|   static_cast<RealType>(100) ),  // Number of failures, k
 | |
|   static_cast<RealType>(9.9000000000003448e-201L), //
 | |
|   100 * tolerance); // Note difference
 | |
| 
 | |
|     // p nearer unity.
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.9999)),
 | |
|   static_cast<RealType>(10) ),  // Number of failures, k
 | |
|   // static_cast<double>(9.9989999999889024e-41), // Boost.Math
 | |
|   // static_cast<float>(1.00156406e-040)
 | |
|   static_cast<RealType>(9.999e-41), // exact from 100 digit calculator.
 | |
|   2e3 * tolerance); // Note bigger tolerance needed.
 | |
| 
 | |
|   // Moshier Cephes 100 digits calculator says 9.999e-41
 | |
|   //0.9999*pow(1-0.9999,10)
 | |
|   // 9.9990000000000000000000000000000000000000000000000000000000000000000000E-41
 | |
|   // 9.998999999988988e-041
 | |
|   // > formatC(dgeom(10, 0.9999), digits=17) [1] "9.9989999999889024e-41"
 | |
|   // p *  pow(q, k)         9.9989999999889880e-041
 | |
|   // exp(p * k * log1p(-p)) 9.9989999999889024e-041
 | |
| 
 | |
| 
 | |
| 
 | |
|   // 0.9999999999 * pow(1-0.9999999999,10)=  9.9999999990E-101
 | |
|   // > formatC(dgeom(10,0.9999999999), digits=17)  [1] "1.0000008273040127e-100"
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.9999999999L)),
 | |
|   static_cast<RealType>(10) ),  //
 | |
|   static_cast<RealType>(9.9999999990E-101L), // 1.0000008273040179e-100
 | |
|   1e9 * tolerance); // Note big tolerance needed.
 | |
|   // 1.0000008273040179e-100  Boost.Math
 | |
|   // 1.0000008273040127e-100  R
 | |
|   // 0.9999999990000004e-100  100 digit calculator 'exact'
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.00000000001L)),
 | |
|   static_cast<RealType>(10) ),  //
 | |
|   static_cast<RealType>(9.999999999e-12L), // get 9.9999999989999994e-012
 | |
|   1 * tolerance); // Note small tolerance needed.
 | |
| 
 | |
| 
 | |
|     BOOST_CHECK_CLOSE_FRACTION( //
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.00000000001L)),
 | |
|   static_cast<RealType>(1000) ),  //
 | |
|   static_cast<RealType>(9.9999999e-12L), // get 9.9999998999999913e-012
 | |
|   tolerance); // Note small tolerance needed.
 | |
| 
 | |
| 
 | |
|   ///////////////////////////////////////////////////
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //
 | |
|     // > formatC(dgeom(0.0001,0.5, FALSE), digits=17) [1] "               0.5"
 | |
|     //  R treates geom as a discrete distribution.
 | |
|     // But Boost.Math is continuous, so if you want R behaviour,
 | |
|     // make number of failures, k into an integer with the floor function.
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.5)),
 | |
|   static_cast<RealType>(floor(0.0001L)) ),  // Number of failures, k is very small but MADE integral,
 | |
|   static_cast<RealType>(0.5), // nearly success probability.
 | |
|   tolerance);
 | |
| 
 | |
|   // R switches over at about 1e7 from k = 0, returning 0.5,  to k = 1, returning 0.25.
 | |
|   // Boost.Math does not do this, even for 0.9999999999999999
 | |
|   // > formatC(pgeom(0.999999,0.5, FALSE), digits=17) [1] "               0.5"
 | |
|   // > formatC(pgeom(0.9999999,0.5, FALSE), digits=17) [1] "              0.25"
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( // > formatC(pgeom(0.0001,0.5, TRUE), digits=17)[1] "               0.5"
 | |
|     // > formatC(pgeom(0.0001,0.5, FALSE), digits=17) [1] "               0.5"
 | |
|     //  R treates geom as a discrete distribution.
 | |
|     // But Boost.Math is continuous, so if you want R behaviour,
 | |
|     // make number of failures, k into an integer with the floor function.
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.5)),
 | |
|   static_cast<RealType>(floor(0.9999999999999999L)) ),  // Number of failures, k is very small but MADE integral,
 | |
|   static_cast<RealType>(0.5), // nearly success probability.
 | |
|   tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( // > formatC(pgeom(0.0001,0.5, TRUE), digits=17)[1] "               0.5"
 | |
|     // > formatC(pgeom(0.0001,0.5, FALSE), digits=17) [1] "               0.5"
 | |
|     //  R treates geom as a discrete distribution.
 | |
|     // But Boost.Math is continuous, so if you want R behaviour,
 | |
|     // make number of failures, k into an integer with the floor function.
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.5)),
 | |
|   static_cast<RealType>(floor(1. - tolerance)) ),
 | |
|   // Number of failures, k is very small but MADE integral,
 | |
|   // Need to use tolerance here,
 | |
|   // as epsilon is ill-defined for Real concept:
 | |
|   // numeric_limits<RealType>::epsilon()  0
 | |
|   static_cast<RealType>(0.5), // nearly success probability.
 | |
|   tolerance * 10);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.0001L)),
 | |
|   static_cast<RealType>(2)),  // k = 2.
 | |
|   static_cast<RealType>(9.99800010e-5L), // 'exact '
 | |
|   tolerance);
 | |
| 
 | |
|   //> formatC(dgeom(2, 0.9999), digits=17) [1] "9.9989999999977806e-09"
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.9999L)),
 | |
|   static_cast<RealType>(2)),  // k = 0
 | |
|   static_cast<RealType>(9.999e-9L), // 'exact'
 | |
|   1000*tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.9999L)),
 | |
|   static_cast<RealType>(3)),  // k = 3
 | |
|   static_cast<RealType>(9.999e-13L), // get
 | |
|   1000*tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   pdf(geometric_distribution<RealType>(static_cast<RealType>(0.9999L)),
 | |
|   static_cast<RealType>(5)),  // k = 5
 | |
|   static_cast<RealType>(9.999e-21L), //  9.9989999999944947e-021
 | |
|   1000*tolerance);
 | |
| 
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   pdf(geometric_distribution<RealType>( static_cast<RealType>(0.0001L)),
 | |
|   static_cast<RealType>(3)),  // k = 0.
 | |
|   static_cast<RealType>(9.99700029999e-5L), //
 | |
|   tolerance);
 | |
|    // Tests on cdf:
 | |
|   // MathCAD pgeom k, r, p) == failures, successes, probability.
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(cdf(
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.5)), // prob 0.5
 | |
|     static_cast<RealType>(0) ), // k = 0
 | |
|     static_cast<RealType>(0.5), // probability =p
 | |
|     tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(cdf(complement(
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.5)), //
 | |
|     static_cast<RealType>(0) )), // k = 0
 | |
|     static_cast<RealType>(0.5), // probability =
 | |
|     tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(cdf(
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.25)), // prob 0.5
 | |
|     static_cast<RealType>(1) ), // k = 0
 | |
|     static_cast<RealType>(0.4375L), // probability =p
 | |
|     tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(cdf(complement(
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.25)), //
 | |
|     static_cast<RealType>(1) )), // k = 0
 | |
|     static_cast<RealType>(1-0.4375L), // probability =
 | |
|     tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(cdf(complement(
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.5)), //
 | |
|     static_cast<RealType>(1) )), // k = 0
 | |
|     static_cast<RealType>(0.25), // probability = exact 0.25
 | |
|     tolerance);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION( //
 | |
|     cdf(geometric_distribution<RealType>(static_cast<RealType>(0.5)),
 | |
|     static_cast<RealType>(4)),  // k =4.
 | |
|     static_cast<RealType>(0.96875L), // exact
 | |
|     tolerance);
 | |
| 
 | |
| 
 | |
|   // Tests of other functions, mean and other moments ...
 | |
| 
 | |
|   geometric_distribution<RealType> dist(static_cast<RealType>(0.25));
 | |
|   // mean:
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|     mean(dist), static_cast<RealType>((1 - 0.25) /0.25), tol5eps);
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|     mode(dist), static_cast<RealType>(0), tol1eps);
 | |
|   // variance:
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|     variance(dist), static_cast<RealType>((1 - 0.25) / (0.25 * 0.25)), tol5eps);
 | |
| 
 | |
|   // std deviation:
 | |
|   // sqrt(0.75/0.125)
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|     standard_deviation(dist), //
 | |
|     static_cast<RealType>(sqrt((1.0L - 0.25L) / (0.25L * 0.25L))), // using 100 digit calc
 | |
|     tol5eps);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|     skewness(dist), //
 | |
|     static_cast<RealType>((2-0.25L) /sqrt(0.75L)),
 | |
|     // using calculator
 | |
|     tol5eps);
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|     kurtosis_excess(dist), //
 | |
|     static_cast<RealType>(6 + 0.0625L/0.75L), //
 | |
|     tol5eps);
 | |
|   // 6.083333333333333  6.166666666666667
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|     kurtosis(dist), // true
 | |
|     static_cast<RealType>(9 + 0.0625L/0.75L), //
 | |
|     tol5eps);
 | |
|   // hazard:
 | |
|   RealType x = static_cast<RealType>(0.125);
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   hazard(dist, x)
 | |
|   , pdf(dist, x) / cdf(complement(dist, x)), tol5eps);
 | |
|   // cumulative hazard:
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   chf(dist, x), -log(cdf(complement(dist, x))), tol5eps);
 | |
|   // coefficient_of_variation:
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   coefficient_of_variation(dist)
 | |
|   , standard_deviation(dist) / mean(dist), tol5eps);
 | |
| 
 | |
|   // Special cases for PDF:
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   pdf(
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0)), //
 | |
|   static_cast<RealType>(0)),
 | |
|   static_cast<RealType>(0) );
 | |
| 
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   pdf(
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0)),
 | |
|   static_cast<RealType>(0.0001)),
 | |
|   static_cast<RealType>(0) );
 | |
| 
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   pdf(
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(1)),
 | |
|   static_cast<RealType>(0.001)),
 | |
|   static_cast<RealType>(0) );
 | |
| 
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   pdf(
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(1)),
 | |
|   static_cast<RealType>(8)),
 | |
|   static_cast<RealType>(0) );
 | |
| 
 | |
|   BOOST_CHECK_SMALL(
 | |
|   pdf(
 | |
|    geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(0))-
 | |
|   static_cast<RealType>(0.25),
 | |
|   2 * boost::math::tools::epsilon<RealType>() ); // Expect exact, but not quite.
 | |
|   // numeric_limits<RealType>::epsilon()); // Not suitable for real concept!
 | |
| 
 | |
|   // Quantile boundary cases checks:
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   quantile(  // zero P < cdf(0) so should be exactly zero.
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(0)),
 | |
|   static_cast<RealType>(0));
 | |
| 
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   quantile(  // min P < cdf(0) so should be exactly zero.
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(boost::math::tools::min_value<RealType>())),
 | |
|   static_cast<RealType>(0));
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   quantile(  // Small P < cdf(0) so should be near zero.
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(boost::math::tools::epsilon<RealType>())), //
 | |
|   static_cast<RealType>(0),
 | |
|     tol5eps);
 | |
| 
 | |
|   BOOST_CHECK_CLOSE_FRACTION(
 | |
|   quantile(  // Small P < cdf(0) so should be exactly zero.
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(0.0001)),
 | |
|   static_cast<RealType>(0),
 | |
|     tolerance);
 | |
| 
 | |
|   //BOOST_CHECK(  // Fails with overflow for real_concept
 | |
|   //quantile(  // Small P near 1 so k failures should be big.
 | |
|   //geometric_distribution<RealType>(static_cast<RealType>(8), static_cast<RealType>(0.25)),
 | |
|   //static_cast<RealType>(1 - boost::math::tools::epsilon<RealType>())) <=
 | |
|   //static_cast<RealType>(189.56999032670058)  // 106.462769 for float
 | |
|   //);
 | |
| 
 | |
|   if(std::numeric_limits<RealType>::has_infinity)
 | |
|   { // BOOST_CHECK tests for infinity using std::numeric_limits<>::infinity()
 | |
|     // Note that infinity is not implemented for real_concept, so these tests
 | |
|     // are only done for types, like built-in float, double.. that have infinity.
 | |
|     // Note that these assume that  BOOST_MATH_OVERFLOW_ERROR_POLICY is NOT throw_on_error.
 | |
|     // #define BOOST_MATH_THROW_ON_OVERFLOW_POLICY ==  throw_on_error would throw here.
 | |
|     // #define BOOST_MAT_DOMAIN_ERROR_POLICY IS defined throw_on_error,
 | |
|     //  so the throw path of error handling is tested below with BOOST_MATH_CHECK_THROW tests.
 | |
| 
 | |
|     BOOST_CHECK(
 | |
|     quantile(  // At P == 1 so k failures should be infinite.
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|     static_cast<RealType>(1)) ==
 | |
|     //static_cast<RealType>(boost::math::tools::infinity<RealType>())
 | |
|     static_cast<RealType>(std::numeric_limits<RealType>::infinity()) );
 | |
| 
 | |
|     BOOST_CHECK_EQUAL(
 | |
|     quantile(  // At 1 == P  so should be infinite.
 | |
|     geometric_distribution<RealType>( static_cast<RealType>(0.25)),
 | |
|     static_cast<RealType>(1)), //
 | |
|     std::numeric_limits<RealType>::infinity() );
 | |
| 
 | |
|     BOOST_CHECK_EQUAL(
 | |
|     quantile(complement(  // Q zero 1 so P == 1 < cdf(0) so should be exactly infinity.
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|     static_cast<RealType>(0))),
 | |
|     std::numeric_limits<RealType>::infinity() );
 | |
|    } // test for infinity using std::numeric_limits<>::infinity()
 | |
|   else
 | |
|   { // real_concept case, so check it throws rather than returning infinity.
 | |
|     BOOST_CHECK_EQUAL(
 | |
|     quantile(  // At P == 1 so k failures should be infinite.
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|     static_cast<RealType>(1)),
 | |
|     boost::math::tools::max_value<RealType>() );
 | |
| 
 | |
|     BOOST_CHECK_EQUAL(
 | |
|     quantile(complement(  // Q zero 1 so P == 1 < cdf(0) so should be exactly infinity.
 | |
|     geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|     static_cast<RealType>(0))),
 | |
|     boost::math::tools::max_value<RealType>());
 | |
|   } // has infinity
 | |
| 
 | |
|   BOOST_CHECK( // Should work for built-in and real_concept.
 | |
|   quantile(complement(  // Q near to 1 so P nearly 1, so should be large > 300.
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(boost::math::tools::min_value<RealType>())))
 | |
|    >= static_cast<RealType>(300) );
 | |
| 
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   quantile(  //  P ==  0 < cdf(0) so should be zero.
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(0)),
 | |
|   static_cast<RealType>(0));
 | |
| 
 | |
|   // Quantile Complement boundary cases:
 | |
| 
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   quantile(complement(  // Q = 1 so P = 0 < cdf(0) so should be exactly zero.
 | |
|   geometric_distribution<RealType>( static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(1))),
 | |
|   static_cast<RealType>(0)
 | |
|   );
 | |
| 
 | |
|   BOOST_CHECK_EQUAL(
 | |
|   quantile(complement(  // Q very near 1 so P == epsilon < cdf(0) so should be exactly zero.
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(1 - boost::math::tools::epsilon<RealType>()))),
 | |
|   static_cast<RealType>(0)
 | |
|   );
 | |
| 
 | |
|   // Check that duff arguments throw domain_error:
 | |
| 
 | |
|   BOOST_MATH_CHECK_THROW(
 | |
|   pdf( // Negative success_fraction!
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(-0.25)),
 | |
|   static_cast<RealType>(0)), std::domain_error);
 | |
|   BOOST_MATH_CHECK_THROW(
 | |
|   pdf( // Success_fraction > 1!
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(1.25)),
 | |
|   static_cast<RealType>(0)),
 | |
|   std::domain_error);
 | |
|   BOOST_MATH_CHECK_THROW(
 | |
|   pdf( // Negative k argument !
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(-1)),
 | |
|   std::domain_error);
 | |
|   //BOOST_MATH_CHECK_THROW(
 | |
|   //pdf( // check limit on k (failures)
 | |
|   //geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   //std::numeric_limits<RealType>infinity()),
 | |
|   //std::domain_error);
 | |
|   BOOST_MATH_CHECK_THROW(
 | |
|   cdf(  // Negative k argument !
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(0.25)),
 | |
|   static_cast<RealType>(-1)),
 | |
|   std::domain_error);
 | |
|   BOOST_MATH_CHECK_THROW(
 | |
|   cdf( // Negative success_fraction!
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(-0.25)),
 | |
|   static_cast<RealType>(0)), std::domain_error);
 | |
|   BOOST_MATH_CHECK_THROW(
 | |
|   cdf( // Success_fraction > 1!
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(1.25)),
 | |
|   static_cast<RealType>(0)), std::domain_error);
 | |
|   BOOST_MATH_CHECK_THROW(
 | |
|   quantile(  // Negative success_fraction!
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(-0.25)),
 | |
|   static_cast<RealType>(0)), std::domain_error);
 | |
|   BOOST_MATH_CHECK_THROW(
 | |
|   quantile( // Success_fraction > 1!
 | |
|   geometric_distribution<RealType>(static_cast<RealType>(1.25)),
 | |
|   static_cast<RealType>(0)), std::domain_error);
 | |
|    check_out_of_range<geometric_distribution<RealType> >(0.5);
 | |
|   // End of check throwing 'duff' out-of-domain values.
 | |
| 
 | |
|   { // Compare geometric and negative binomial functions.
 | |
|     using boost::math::negative_binomial_distribution;
 | |
|     using boost::math::geometric_distribution;
 | |
| 
 | |
|     RealType k = static_cast<RealType>(2.L);
 | |
|     RealType alpha = static_cast<RealType>(0.05L);
 | |
|     RealType p = static_cast<RealType>(0.5L);
 | |
| 
 | |
|     BOOST_CHECK_CLOSE_FRACTION( // Successes parameter in negative binomial is 1 for geometric.
 | |
|       geometric_distribution<RealType>::find_lower_bound_on_p(k, alpha),
 | |
|       negative_binomial_distribution<RealType>::find_lower_bound_on_p(k, static_cast<RealType>(1), alpha),
 | |
|       tolerance);
 | |
|     BOOST_CHECK_CLOSE_FRACTION( // Successes parameter in negative binomial is 1 for geometric.
 | |
|       geometric_distribution<RealType>::find_upper_bound_on_p(k, alpha),
 | |
|       negative_binomial_distribution<RealType>::find_upper_bound_on_p(k, static_cast<RealType>(1), alpha),
 | |
|       tolerance);
 | |
|     BOOST_CHECK_CLOSE_FRACTION( // Should be identical - successes parameter is not used.
 | |
|        geometric_distribution<RealType>::find_maximum_number_of_trials(k, p, alpha),
 | |
|       negative_binomial_distribution<RealType>::find_maximum_number_of_trials(k, p, alpha),
 | |
|     tolerance);
 | |
|   }
 | |
|     //geometric::find_upper_bound_on_p(k, alpha);
 | |
|    return;
 | |
| } // template <class RealType> void test_spots(RealType) // Any floating-point type RealType.
 | |
| 
 | |
| BOOST_AUTO_TEST_CASE( test_main )
 | |
| {
 | |
|   // Check that can generate geometric distribution using the two convenience methods:
 | |
|    using namespace boost::math;
 | |
|    geometric g05d(0.5); // Using typedef - default type is double.
 | |
|    geometric_distribution<> g05dd(0.5); // Using default RealType double.
 | |
| 
 | |
|   // Basic sanity-check spot values.
 | |
| 
 | |
|   // Test some simple double only examples.
 | |
|   geometric_distribution<double> mydist(0.25);
 | |
|   // success fraction == 0.25 == 25% or 1 in 4 successes.
 | |
|   // Note: double values (matching the distribution definition) avoid the need for any casting.
 | |
| 
 | |
|   // Check accessor functions return exact values for double at least.
 | |
|   BOOST_CHECK_EQUAL(mydist.success_fraction(), static_cast<double>(1./4.));
 | |
| 
 | |
|   //cout << numeric_limits<RealType>::epsilon() << endl;
 | |
| 
 | |
|   // (Parameter value, arbitrarily zero, only communicates the floating point type).
 | |
| #ifdef TEST_FLOAT
 | |
|   test_spots(0.0F); // Test float.
 | |
| #endif
 | |
| #ifdef TEST_DOUBLE
 | |
|   test_spots(0.0); // Test double.
 | |
| #endif
 | |
| #ifndef BOOST_MATH_NO_LONG_DOUBLE_MATH_FUNCTIONS
 | |
| #ifdef TEST_LDOUBLE
 | |
|   test_spots(0.0L); // Test long double.
 | |
| #endif
 | |
|   #if !BOOST_WORKAROUND(__BORLANDC__, BOOST_TESTED_AT(0x582))
 | |
| #ifdef TEST_REAL_CONCEPT
 | |
|     test_spots(boost::math::concepts::real_concept(0.)); // Test real concept.
 | |
| #endif
 | |
|   #endif
 | |
| #else
 | |
|    std::cout << "<note>The long double tests have been disabled on this platform "
 | |
|       "either because the long double overloads of the usual math functions are "
 | |
|       "not available at all, or because they are too inaccurate for these tests "
 | |
|       "to pass.</note>" << std::endl;
 | |
| #endif
 | |
| 
 | |
|   
 | |
| } // BOOST_AUTO_TEST_CASE( test_main )
 | |
| 
 | |
| /*
 | |
| 
 | |
| 
 | |
| 
 | |
| */
 |