Blame randist/test.c

Packit 67cb25
/* randist/test.c
Packit 67cb25
 * 
Packit 67cb25
 * Copyright (C) 1996, 1997, 1998, 1999, 2000, 2007, 2010 James Theiler, Brian Gough
Packit 67cb25
 * 
Packit 67cb25
 * This program is free software; you can redistribute it and/or modify
Packit 67cb25
 * it under the terms of the GNU General Public License as published by
Packit 67cb25
 * the Free Software Foundation; either version 3 of the License, or (at
Packit 67cb25
 * your option) any later version.
Packit 67cb25
 * 
Packit 67cb25
 * This program is distributed in the hope that it will be useful, but
Packit 67cb25
 * WITHOUT ANY WARRANTY; without even the implied warranty of
Packit 67cb25
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Packit 67cb25
 * General Public License for more details.
Packit 67cb25
 * 
Packit 67cb25
 * You should have received a copy of the GNU General Public License
Packit 67cb25
 * along with this program; if not, write to the Free Software
Packit 67cb25
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
Packit 67cb25
 */
Packit 67cb25
Packit 67cb25
#include <config.h>
Packit 67cb25
#include <stdio.h>
Packit 67cb25
#include <stdlib.h>
Packit 67cb25
#include <math.h>
Packit 67cb25
#include <gsl/gsl_math.h>
Packit 67cb25
#include <gsl/gsl_randist.h>
Packit 67cb25
#include <gsl/gsl_rng.h>
Packit 67cb25
#include <gsl/gsl_test.h>
Packit 67cb25
#include <gsl/gsl_ieee_utils.h>
Packit 67cb25
#include <gsl/gsl_integration.h>
Packit 67cb25
#include <gsl/gsl_vector.h>
Packit 67cb25
#include <gsl/gsl_matrix.h>
Packit 67cb25
#include <gsl/gsl_blas.h>
Packit 67cb25
#include <gsl/gsl_linalg.h>
Packit 67cb25
#include <gsl/gsl_cdf.h>
Packit 67cb25
#include <gsl/gsl_statistics.h>
Packit 67cb25
Packit 67cb25
#define N 100000
Packit 67cb25
Packit 67cb25
/* Convient test dimension for multivariant distributions */
Packit 67cb25
#define MULTI_DIM 10
Packit 67cb25
Packit 67cb25
Packit 67cb25
void testMoments (double (*f) (void), const char *name,
Packit 67cb25
                  double a, double b, double p);
Packit 67cb25
void testPDF (double (*f) (void), double (*pdf) (double), const char *name);
Packit 67cb25
void testDiscretePDF (double (*f) (void), double (*pdf) (unsigned int),
Packit 67cb25
                      const char *name);
Packit 67cb25
Packit 67cb25
void test_shuffle (void);
Packit 67cb25
void test_choose (void);
Packit 67cb25
double test_beta (void);
Packit 67cb25
double test_beta_pdf (double x);
Packit 67cb25
double test_bernoulli (void);
Packit 67cb25
double test_bernoulli_pdf (unsigned int n);
Packit 67cb25
Packit 67cb25
double test_binomial (void);
Packit 67cb25
double test_binomial_pdf (unsigned int n);
Packit 67cb25
double test_binomial_large (void);
Packit 67cb25
double test_binomial_large_pdf (unsigned int n);
Packit 67cb25
double test_binomial_huge (void);
Packit 67cb25
double test_binomial_huge_pdf (unsigned int n);
Packit 67cb25
double test_binomial_max (void);
Packit 67cb25
double test_binomial_max_pdf (unsigned int n);
Packit 67cb25
double test_binomial0 (void);
Packit 67cb25
double test_binomial0_pdf (unsigned int n);
Packit 67cb25
double test_binomial1 (void);
Packit 67cb25
double test_binomial1_pdf (unsigned int n);
Packit 67cb25
Packit 67cb25
Packit 67cb25
Packit 67cb25
double test_binomial_knuth (void);
Packit 67cb25
double test_binomial_knuth_pdf (unsigned int n);
Packit 67cb25
double test_binomial_large_knuth (void);
Packit 67cb25
double test_binomial_large_knuth_pdf (unsigned int n);
Packit 67cb25
double test_binomial_huge_knuth (void);
Packit 67cb25
double test_binomial_huge_knuth_pdf (unsigned int n);
Packit 67cb25
Packit 67cb25
double test_cauchy (void);
Packit 67cb25
double test_cauchy_pdf (double x);
Packit 67cb25
double test_chisq (void);
Packit 67cb25
double test_chisq_pdf (double x);
Packit 67cb25
double test_chisqnu2 (void);
Packit 67cb25
double test_chisqnu2_pdf (double x);
Packit 67cb25
double test_dirichlet (void);
Packit 67cb25
double test_dirichlet_pdf (double x);
Packit 67cb25
double test_dirichlet_small (void);
Packit 67cb25
double test_dirichlet_small_pdf (double x);
Packit 67cb25
void test_dirichlet_moments (void);
Packit 67cb25
double test_discrete1 (void);
Packit 67cb25
double test_discrete1_pdf (unsigned int n);
Packit 67cb25
double test_discrete2 (void);
Packit 67cb25
double test_discrete2_pdf (unsigned int n);
Packit 67cb25
double test_discrete3 (void);
Packit 67cb25
double test_discrete3_pdf (unsigned int n);
Packit 67cb25
double test_erlang (void);
Packit 67cb25
double test_erlang_pdf (double x);
Packit 67cb25
double test_exponential (void);
Packit 67cb25
double test_exponential_pdf (double x);
Packit 67cb25
double test_exppow0 (void);
Packit 67cb25
double test_exppow0_pdf (double x);
Packit 67cb25
double test_exppow1 (void);
Packit 67cb25
double test_exppow1_pdf (double x);
Packit 67cb25
double test_exppow1a (void);
Packit 67cb25
double test_exppow1a_pdf (double x);
Packit 67cb25
double test_exppow2 (void);
Packit 67cb25
double test_exppow2_pdf (double x);
Packit 67cb25
double test_exppow2a (void);
Packit 67cb25
double test_exppow2a_pdf (double x);
Packit 67cb25
double test_exppow2b (void);
Packit 67cb25
double test_exppow2b_pdf (double x);
Packit 67cb25
double test_fdist (void);
Packit 67cb25
double test_fdist_pdf (double x);
Packit 67cb25
double test_fdist_large (void);
Packit 67cb25
double test_fdist_large_pdf (double x);
Packit 67cb25
double test_flat (void);
Packit 67cb25
double test_flat_pdf (double x);
Packit 67cb25
double test_gamma (void);
Packit 67cb25
double test_gamma_pdf (double x);
Packit 67cb25
double test_gamma1 (void);
Packit 67cb25
double test_gamma1_pdf (double x);
Packit 67cb25
double test_gamma_int (void);
Packit 67cb25
double test_gamma_int_pdf (double x);
Packit 67cb25
double test_gamma_large (void);
Packit 67cb25
double test_gamma_large_pdf (double x);
Packit 67cb25
double test_gamma_vlarge (void);
Packit 67cb25
double test_gamma_vlarge_pdf (double x);
Packit 67cb25
double test_gamma_small (void);
Packit 67cb25
double test_gamma_small_pdf (double x);
Packit 67cb25
double test_gamma_mt (void);
Packit 67cb25
double test_gamma_mt_pdf (double x);
Packit 67cb25
double test_gamma_mt1 (void);
Packit 67cb25
double test_gamma_mt1_pdf (double x);
Packit 67cb25
double test_gamma_mt_int (void);
Packit 67cb25
double test_gamma_mt_int_pdf (double x);
Packit 67cb25
double test_gamma_mt_large (void);
Packit 67cb25
double test_gamma_mt_large_pdf (double x);
Packit 67cb25
double test_gamma_mt_small (void);
Packit 67cb25
double test_gamma_mt_small_pdf (double x);
Packit 67cb25
double test_gamma_knuth_vlarge (void);
Packit 67cb25
double test_gamma_knuth_vlarge_pdf (double x);
Packit 67cb25
double test_gaussian (void);
Packit 67cb25
double test_gaussian_pdf (double x);
Packit 67cb25
double test_gaussian_ratio_method (void);
Packit 67cb25
double test_gaussian_ratio_method_pdf (double x);
Packit 67cb25
double test_gaussian_ziggurat (void);
Packit 67cb25
double test_gaussian_ziggurat_pdf (double x);
Packit 67cb25
double test_gaussian_tail (void);
Packit 67cb25
double test_gaussian_tail_pdf (double x);
Packit 67cb25
double test_gaussian_tail1 (void);
Packit 67cb25
double test_gaussian_tail1_pdf (double x);
Packit 67cb25
double test_gaussian_tail2 (void);
Packit 67cb25
double test_gaussian_tail2_pdf (double x);
Packit 67cb25
double test_ugaussian (void);
Packit 67cb25
double test_ugaussian_pdf (double x);
Packit 67cb25
double test_ugaussian_ratio_method (void);
Packit 67cb25
double test_ugaussian_ratio_method_pdf (double x);
Packit 67cb25
double test_ugaussian_tail (void);
Packit 67cb25
double test_ugaussian_tail_pdf (double x);
Packit 67cb25
double test_bivariate_gaussian1 (void);
Packit 67cb25
double test_bivariate_gaussian1_pdf (double x);
Packit 67cb25
double test_bivariate_gaussian2 (void);
Packit 67cb25
double test_bivariate_gaussian2_pdf (double x);
Packit 67cb25
double test_bivariate_gaussian3 (void);
Packit 67cb25
double test_bivariate_gaussian3_pdf (double x);
Packit 67cb25
double test_bivariate_gaussian4 (void);
Packit 67cb25
double test_bivariate_gaussian4_pdf (double x);
Packit 67cb25
void test_multivariate_gaussian_log_pdf (void);
Packit 67cb25
void test_multivariate_gaussian_pdf (void);
Packit 67cb25
void test_multivariate_gaussian (void);
Packit 67cb25
void test_wishart_log_pdf (void);
Packit 67cb25
void test_wishart_pdf (void);
Packit 67cb25
void test_wishart (void);
Packit 67cb25
double test_gumbel1 (void);
Packit 67cb25
double test_gumbel1_pdf (double x);
Packit 67cb25
double test_gumbel2 (void);
Packit 67cb25
double test_gumbel2_pdf (double x);
Packit 67cb25
double test_geometric (void);
Packit 67cb25
double test_geometric_pdf (unsigned int x);
Packit 67cb25
double test_geometric1 (void);
Packit 67cb25
double test_geometric1_pdf (unsigned int x);
Packit 67cb25
double test_hypergeometric1 (void);
Packit 67cb25
double test_hypergeometric1_pdf (unsigned int x);
Packit 67cb25
double test_hypergeometric2 (void);
Packit 67cb25
double test_hypergeometric2_pdf (unsigned int x);
Packit 67cb25
double test_hypergeometric3 (void);
Packit 67cb25
double test_hypergeometric3_pdf (unsigned int x);
Packit 67cb25
double test_hypergeometric4 (void);
Packit 67cb25
double test_hypergeometric4_pdf (unsigned int x);
Packit 67cb25
double test_hypergeometric5 (void);
Packit 67cb25
double test_hypergeometric5_pdf (unsigned int x);
Packit 67cb25
double test_hypergeometric6 (void);
Packit 67cb25
double test_hypergeometric6_pdf (unsigned int x);
Packit 67cb25
double test_landau (void);
Packit 67cb25
double test_landau_pdf (double x);
Packit 67cb25
double test_levy1 (void);
Packit 67cb25
double test_levy1_pdf (double x);
Packit 67cb25
double test_levy2 (void);
Packit 67cb25
double test_levy2_pdf (double x);
Packit 67cb25
double test_levy1a (void);
Packit 67cb25
double test_levy1a_pdf (double x);
Packit 67cb25
double test_levy2a (void);
Packit 67cb25
double test_levy2a_pdf (double x);
Packit 67cb25
double test_levy_skew1 (void);
Packit 67cb25
double test_levy_skew1_pdf (double x);
Packit 67cb25
double test_levy_skew2 (void);
Packit 67cb25
double test_levy_skew2_pdf (double x);
Packit 67cb25
double test_levy_skew1a (void);
Packit 67cb25
double test_levy_skew1a_pdf (double x);
Packit 67cb25
double test_levy_skew2a (void);
Packit 67cb25
double test_levy_skew2a_pdf (double x);
Packit 67cb25
double test_levy_skew1b (void);
Packit 67cb25
double test_levy_skew1b_pdf (double x);
Packit 67cb25
double test_levy_skew2b (void);
Packit 67cb25
double test_levy_skew2b_pdf (double x);
Packit 67cb25
double test_logistic (void);
Packit 67cb25
double test_logistic_pdf (double x);
Packit 67cb25
double test_lognormal (void);
Packit 67cb25
double test_lognormal_pdf (double x);
Packit 67cb25
double test_logarithmic (void);
Packit 67cb25
double test_logarithmic_pdf (unsigned int n);
Packit 67cb25
double test_multinomial (void);
Packit 67cb25
double test_multinomial_pdf (unsigned int n);
Packit 67cb25
double test_multinomial_large (void);
Packit 67cb25
double test_multinomial_large_pdf (unsigned int n);
Packit 67cb25
void test_multinomial_moments (void);
Packit 67cb25
double test_negative_binomial (void);
Packit 67cb25
double test_negative_binomial_pdf (unsigned int n);
Packit 67cb25
double test_pascal (void);
Packit 67cb25
double test_pascal_pdf (unsigned int n);
Packit 67cb25
double test_pareto (void);
Packit 67cb25
double test_pareto_pdf (double x);
Packit 67cb25
double test_poisson (void);
Packit 67cb25
double test_poisson_pdf (unsigned int x);
Packit 67cb25
double test_poisson_large (void);
Packit 67cb25
double test_poisson_large_pdf (unsigned int x);
Packit 67cb25
double test_dir2d (void);
Packit 67cb25
double test_dir2d_pdf (double x);
Packit 67cb25
double test_dir2d_trig_method (void);
Packit 67cb25
double test_dir2d_trig_method_pdf (double x);
Packit 67cb25
double test_dir3dxy (void);
Packit 67cb25
double test_dir3dxy_pdf (double x);
Packit 67cb25
double test_dir3dyz (void);
Packit 67cb25
double test_dir3dyz_pdf (double x);
Packit 67cb25
double test_dir3dzx (void);
Packit 67cb25
double test_dir3dzx_pdf (double x);
Packit 67cb25
double test_rayleigh (void);
Packit 67cb25
double test_rayleigh_pdf (double x);
Packit 67cb25
double test_rayleigh_tail (void);
Packit 67cb25
double test_rayleigh_tail_pdf (double x);
Packit 67cb25
double test_tdist1 (void);
Packit 67cb25
double test_tdist1_pdf (double x);
Packit 67cb25
double test_tdist2 (void);
Packit 67cb25
double test_tdist2_pdf (double x);
Packit 67cb25
double test_laplace (void);
Packit 67cb25
double test_laplace_pdf (double x);
Packit 67cb25
double test_weibull (void);
Packit 67cb25
double test_weibull_pdf (double x);
Packit 67cb25
double test_weibull1 (void);
Packit 67cb25
double test_weibull1_pdf (double x);
Packit 67cb25
Packit 67cb25
gsl_rng *r_global;
Packit 67cb25
Packit 67cb25
static gsl_ran_discrete_t *g1 = NULL;
Packit 67cb25
static gsl_ran_discrete_t *g2 = NULL;
Packit 67cb25
static gsl_ran_discrete_t *g3 = NULL;
Packit 67cb25
Packit 67cb25
int
Packit 67cb25
main (void)
Packit 67cb25
{
Packit 67cb25
  gsl_ieee_env_setup ();
Packit 67cb25
Packit 67cb25
  gsl_rng_env_setup ();
Packit 67cb25
  r_global = gsl_rng_alloc (gsl_rng_default);
Packit 67cb25
Packit 67cb25
#define FUNC(x)  test_ ## x,                     "test gsl_ran_" #x
Packit 67cb25
#define FUNC2(x) test_ ## x, test_ ## x ## _pdf, "test gsl_ran_" #x
Packit 67cb25
Packit 67cb25
  test_shuffle ();
Packit 67cb25
  test_choose ();
Packit 67cb25
Packit 67cb25
  testMoments (FUNC (ugaussian), 0.0, 100.0, 0.5);
Packit 67cb25
  testMoments (FUNC (ugaussian), -1.0, 1.0, 0.6826895);
Packit 67cb25
  testMoments (FUNC (ugaussian), 3.0, 3.5, 0.0011172689);
Packit 67cb25
  testMoments (FUNC (ugaussian_tail), 3.0, 3.5, 0.0011172689 / 0.0013498981);
Packit 67cb25
  testMoments (FUNC (exponential), 0.0, 1.0, 1 - exp (-0.5));
Packit 67cb25
  testMoments (FUNC (cauchy), 0.0, 10000.0, 0.5);
Packit 67cb25
Packit 67cb25
  testMoments (FUNC (discrete1), -0.5, 0.5, 0.59);
Packit 67cb25
  testMoments (FUNC (discrete1), 0.5, 1.5, 0.40);
Packit 67cb25
  testMoments (FUNC (discrete1), 1.5, 3.5, 0.01);
Packit 67cb25
Packit 67cb25
  testMoments (FUNC (discrete2), -0.5,  0.5, 1.0/45.0 );
Packit 67cb25
  testMoments (FUNC (discrete2),  8.5,  9.5, 0 );
Packit 67cb25
  
Packit 67cb25
  testMoments (FUNC (discrete3), -0.5, 0.5, 0.05 );
Packit 67cb25
  testMoments (FUNC (discrete3),  0.5, 1.5, 0.05 );
Packit 67cb25
  testMoments (FUNC (discrete3), -0.5, 9.5, 0.5 );
Packit 67cb25
Packit 67cb25
  test_dirichlet_moments ();
Packit 67cb25
  test_multinomial_moments ();
Packit 67cb25
Packit 67cb25
  testPDF (FUNC2 (beta));
Packit 67cb25
  testPDF (FUNC2 (cauchy));
Packit 67cb25
  testPDF (FUNC2 (chisq));
Packit 67cb25
  testPDF (FUNC2 (chisqnu2));
Packit 67cb25
  testPDF (FUNC2 (dirichlet));
Packit 67cb25
  testPDF (FUNC2 (dirichlet_small));
Packit 67cb25
  testPDF (FUNC2 (erlang));
Packit 67cb25
  testPDF (FUNC2 (exponential));
Packit 67cb25
Packit 67cb25
  testPDF (FUNC2 (exppow0));
Packit 67cb25
  testPDF (FUNC2 (exppow1));
Packit 67cb25
  testPDF (FUNC2 (exppow1a));
Packit 67cb25
  testPDF (FUNC2 (exppow2));
Packit 67cb25
  testPDF (FUNC2 (exppow2a));
Packit 67cb25
  testPDF (FUNC2 (exppow2b));
Packit 67cb25
Packit 67cb25
  testPDF (FUNC2 (fdist));
Packit 67cb25
  testPDF (FUNC2 (fdist_large));
Packit 67cb25
  testPDF (FUNC2 (flat));
Packit 67cb25
  testPDF (FUNC2 (gamma));
Packit 67cb25
  testPDF (FUNC2 (gamma1));
Packit 67cb25
  testPDF (FUNC2 (gamma_int));
Packit 67cb25
  testPDF (FUNC2 (gamma_large));
Packit 67cb25
  testPDF (FUNC2 (gamma_vlarge));
Packit 67cb25
  testPDF (FUNC2 (gamma_knuth_vlarge));
Packit 67cb25
  testPDF (FUNC2 (gamma_small));
Packit 67cb25
  testPDF (FUNC2 (gamma_mt));
Packit 67cb25
  testPDF (FUNC2 (gamma_mt1));
Packit 67cb25
  testPDF (FUNC2 (gamma_mt_int));
Packit 67cb25
  testPDF (FUNC2 (gamma_mt_large));
Packit 67cb25
  testPDF (FUNC2 (gamma_mt_small));
Packit 67cb25
  testPDF (FUNC2 (gaussian));
Packit 67cb25
  testPDF (FUNC2 (gaussian_ratio_method));
Packit 67cb25
  testPDF (FUNC2 (gaussian_ziggurat));
Packit 67cb25
  testPDF (FUNC2 (ugaussian));
Packit 67cb25
  testPDF (FUNC2 (ugaussian_ratio_method));
Packit 67cb25
  testPDF (FUNC2 (gaussian_tail));
Packit 67cb25
  testPDF (FUNC2 (gaussian_tail1));
Packit 67cb25
  testPDF (FUNC2 (gaussian_tail2));
Packit 67cb25
  testPDF (FUNC2 (ugaussian_tail));
Packit 67cb25
Packit 67cb25
  testPDF (FUNC2 (bivariate_gaussian1));
Packit 67cb25
  testPDF (FUNC2 (bivariate_gaussian2));
Packit 67cb25
  testPDF (FUNC2 (bivariate_gaussian3));
Packit 67cb25
  testPDF (FUNC2 (bivariate_gaussian4));
Packit 67cb25
Packit 67cb25
  test_multivariate_gaussian_log_pdf ();
Packit 67cb25
  test_multivariate_gaussian_pdf ();
Packit 67cb25
  test_multivariate_gaussian ();
Packit 67cb25
Packit 67cb25
  test_wishart_log_pdf ();
Packit 67cb25
  test_wishart_pdf ();
Packit 67cb25
  test_wishart ();
Packit 67cb25
Packit 67cb25
  testPDF (FUNC2 (gumbel1));
Packit 67cb25
  testPDF (FUNC2 (gumbel2));
Packit 67cb25
  testPDF (FUNC2 (landau));
Packit 67cb25
  testPDF (FUNC2 (levy1));
Packit 67cb25
  testPDF (FUNC2 (levy2));
Packit 67cb25
  testPDF (FUNC2 (levy1a));
Packit 67cb25
  testPDF (FUNC2 (levy2a));
Packit 67cb25
  testPDF (FUNC2 (levy_skew1));
Packit 67cb25
  testPDF (FUNC2 (levy_skew2));
Packit 67cb25
  testPDF (FUNC2 (levy_skew1a));
Packit 67cb25
  testPDF (FUNC2 (levy_skew2a));
Packit 67cb25
  testPDF (FUNC2 (levy_skew1b));
Packit 67cb25
  testPDF (FUNC2 (levy_skew2b));
Packit 67cb25
  testPDF (FUNC2 (logistic));
Packit 67cb25
  testPDF (FUNC2 (lognormal));
Packit 67cb25
  testPDF (FUNC2 (pareto));
Packit 67cb25
  testPDF (FUNC2 (rayleigh));
Packit 67cb25
  testPDF (FUNC2 (rayleigh_tail));
Packit 67cb25
  testPDF (FUNC2 (tdist1));
Packit 67cb25
  testPDF (FUNC2 (tdist2));
Packit 67cb25
  testPDF (FUNC2 (laplace));
Packit 67cb25
  testPDF (FUNC2 (weibull));
Packit 67cb25
  testPDF (FUNC2 (weibull1));
Packit 67cb25
Packit 67cb25
  testPDF (FUNC2 (dir2d));
Packit 67cb25
  testPDF (FUNC2 (dir2d_trig_method));
Packit 67cb25
  testPDF (FUNC2 (dir3dxy));
Packit 67cb25
  testPDF (FUNC2 (dir3dyz));
Packit 67cb25
  testPDF (FUNC2 (dir3dzx));
Packit 67cb25
Packit 67cb25
  testDiscretePDF (FUNC2 (discrete1));
Packit 67cb25
  testDiscretePDF (FUNC2 (discrete2));
Packit 67cb25
  testDiscretePDF (FUNC2 (discrete3));
Packit 67cb25
  testDiscretePDF (FUNC2 (poisson));
Packit 67cb25
  testDiscretePDF (FUNC2 (poisson_large));
Packit 67cb25
  testDiscretePDF (FUNC2 (bernoulli));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial0));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial1));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial_knuth));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial_large));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial_large_knuth));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial_huge));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial_huge_knuth));
Packit 67cb25
  testDiscretePDF (FUNC2 (binomial_max));
Packit 67cb25
  testDiscretePDF (FUNC2 (geometric));
Packit 67cb25
  testDiscretePDF (FUNC2 (geometric1));
Packit 67cb25
  testDiscretePDF (FUNC2 (hypergeometric1));
Packit 67cb25
  testDiscretePDF (FUNC2 (hypergeometric2));
Packit 67cb25
  testDiscretePDF (FUNC2 (hypergeometric3));
Packit 67cb25
  testDiscretePDF (FUNC2 (hypergeometric4));
Packit 67cb25
  testDiscretePDF (FUNC2 (hypergeometric5));
Packit 67cb25
  testDiscretePDF (FUNC2 (hypergeometric6));
Packit 67cb25
  testDiscretePDF (FUNC2 (logarithmic));
Packit 67cb25
  testDiscretePDF (FUNC2 (multinomial));
Packit 67cb25
  testDiscretePDF (FUNC2 (multinomial_large));
Packit 67cb25
  testDiscretePDF (FUNC2 (negative_binomial));
Packit 67cb25
  testDiscretePDF (FUNC2 (pascal));
Packit 67cb25
Packit 67cb25
  gsl_rng_free (r_global);
Packit 67cb25
  gsl_ran_discrete_free (g1);
Packit 67cb25
  gsl_ran_discrete_free (g2);
Packit 67cb25
  gsl_ran_discrete_free (g3);
Packit 67cb25
Packit 67cb25
  exit (gsl_test_summary ());
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
void
Packit 67cb25
test_shuffle (void)
Packit 67cb25
{
Packit 67cb25
  double count[10][10];
Packit 67cb25
  int x[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
Packit 67cb25
  int i, j, status = 0;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < 10; i++)
Packit 67cb25
    {
Packit 67cb25
      for (j = 0; j < 10; j++)
Packit 67cb25
        {
Packit 67cb25
          count[i][j] = 0;
Packit 67cb25
        }
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  for (i = 0; i < N; i++)
Packit 67cb25
    {
Packit 67cb25
      for (j = 0; j < 10; j++)
Packit 67cb25
        x[j] = j;
Packit 67cb25
Packit 67cb25
      gsl_ran_shuffle (r_global, x, 10, sizeof (int));
Packit 67cb25
Packit 67cb25
      for (j = 0; j < 10; j++)
Packit 67cb25
        count[x[j]][j]++;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  for (i = 0; i < 10; i++)
Packit 67cb25
    {
Packit 67cb25
      for (j = 0; j < 10; j++)
Packit 67cb25
        {
Packit 67cb25
          double expected = N / 10.0;
Packit 67cb25
          double d = fabs (count[i][j] - expected);
Packit 67cb25
          double sigma = d / sqrt (expected);
Packit 67cb25
          if (sigma > 5 && d > 1)
Packit 67cb25
            {
Packit 67cb25
              status = 1;
Packit 67cb25
              gsl_test (status,
Packit 67cb25
                        "gsl_ran_shuffle %d,%d (%g observed vs %g expected)",
Packit 67cb25
                        i, j, count[i][j] / N, 0.1);
Packit 67cb25
            }
Packit 67cb25
        }
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  gsl_test (status, "gsl_ran_shuffle on {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}");
Packit 67cb25
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
void
Packit 67cb25
test_choose (void)
Packit 67cb25
{
Packit 67cb25
  double count[10];
Packit 67cb25
  int x[10] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
Packit 67cb25
  int y[3] = { 0, 1, 2 };
Packit 67cb25
  int i, j, status = 0;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < 10; i++)
Packit 67cb25
    {
Packit 67cb25
      count[i] = 0;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  for (i = 0; i < N; i++)
Packit 67cb25
    {
Packit 67cb25
      for (j = 0; j < 10; j++)
Packit 67cb25
        x[j] = j;
Packit 67cb25
Packit 67cb25
      gsl_ran_choose (r_global, y, 3, x, 10, sizeof (int));
Packit 67cb25
Packit 67cb25
      for (j = 0; j < 3; j++)
Packit 67cb25
        count[y[j]]++;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  for (i = 0; i < 10; i++)
Packit 67cb25
    {
Packit 67cb25
      double expected = 3.0 * N / 10.0;
Packit 67cb25
      double d = fabs (count[i] - expected);
Packit 67cb25
      double sigma = d / sqrt (expected);
Packit 67cb25
      if (sigma > 5 && d > 1)
Packit 67cb25
        {
Packit 67cb25
          status = 1;
Packit 67cb25
          gsl_test (status,
Packit 67cb25
                    "gsl_ran_choose %d (%g observed vs %g expected)",
Packit 67cb25
                    i, count[i] / N, 0.1);
Packit 67cb25
        }
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  gsl_test (status, "gsl_ran_choose (3) on {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}");
Packit 67cb25
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
Packit 67cb25
Packit 67cb25
void
Packit 67cb25
testMoments (double (*f) (void), const char *name,
Packit 67cb25
             double a, double b, double p)
Packit 67cb25
{
Packit 67cb25
  int i;
Packit 67cb25
  double count = 0, expected, sigma;
Packit 67cb25
  int status;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < N; i++)
Packit 67cb25
    {
Packit 67cb25
      double r = f ();
Packit 67cb25
      if (r < b && r > a)
Packit 67cb25
        count++;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  expected = p * N;
Packit 67cb25
  sigma = (expected > 0) ? fabs (count - expected) / sqrt (expected) : fabs(count - expected);
Packit 67cb25
Packit 67cb25
  status = (sigma > 3);
Packit 67cb25
Packit 67cb25
  gsl_test (status, "%s [%g,%g] (%g observed vs %g expected)",
Packit 67cb25
            name, a, b, count / N, p);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
#define BINS 100
Packit 67cb25
Packit 67cb25
typedef double pdf_func(double);
Packit 67cb25
Packit 67cb25
/* Keep track of invalid values during integration */
Packit 67cb25
static int pdf_errors = 0;
Packit 67cb25
static double pdf_errval = 0.0;
Packit 67cb25
Packit 67cb25
double 
Packit 67cb25
wrapper_function (double x, void *params)
Packit 67cb25
{
Packit 67cb25
  pdf_func * pdf = (pdf_func *)params;
Packit 67cb25
  double P = pdf(x);
Packit 67cb25
  if (!gsl_finite(P)) {  
Packit 67cb25
    pdf_errors++;
Packit 67cb25
    pdf_errval = P;
Packit 67cb25
    P = 0; /* skip invalid value now, but return pdf_errval at the end */
Packit 67cb25
  }
Packit 67cb25
  return P;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
integrate (pdf_func * pdf, double a, double b)
Packit 67cb25
{
Packit 67cb25
  double result, abserr;
Packit 67cb25
  size_t n = 1000;
Packit 67cb25
  gsl_function f;  
Packit 67cb25
  gsl_integration_workspace * w = gsl_integration_workspace_alloc (n);
Packit 67cb25
  f.function = &wrapper_function;
Packit 67cb25
  f.params = (void *)pdf;
Packit 67cb25
  pdf_errors = 0;
Packit 67cb25
  gsl_integration_qags (&f, a, b, 1e-16, 1e-4, n, w, &result, &abserr);
Packit 67cb25
  gsl_integration_workspace_free (w);
Packit 67cb25
  if (pdf_errors) return pdf_errval;
Packit 67cb25
  return result;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
void
Packit 67cb25
testPDF (double (*f) (void), double (*pdf) (double), const char *name)
Packit 67cb25
{
Packit 67cb25
  double count[BINS], edge[BINS], p[BINS];
Packit 67cb25
  double a = -5.0, b = +5.0;
Packit 67cb25
  double dx = (b - a) / BINS;
Packit 67cb25
  double bin;
Packit 67cb25
  double total = 0, mean;
Packit 67cb25
  int i, j, status = 0, status_i = 0, attempts = 0;
Packit 67cb25
  long int n0 = 0, n = N;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < BINS; i++)
Packit 67cb25
    {
Packit 67cb25
      /* Compute the integral of p(x) from x to x+dx */
Packit 67cb25
Packit 67cb25
      double x = a + i * dx;
Packit 67cb25
Packit 67cb25
      if (fabs (x) < 1e-10)     /* hit the origin exactly */
Packit 67cb25
        x = 0.0;
Packit 67cb25
Packit 67cb25
      p[i]  = integrate (pdf, x, x+dx);
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
Packit 67cb25
  for (i = 0; i < BINS; i++)
Packit 67cb25
    {
Packit 67cb25
      count[i] = 0;
Packit 67cb25
      edge[i] = 0;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
 trial:
Packit 67cb25
  attempts++;
Packit 67cb25
Packit 67cb25
  for (i = n0; i < n; i++)
Packit 67cb25
    {
Packit 67cb25
      double r = f ();
Packit 67cb25
      total += r;
Packit 67cb25
Packit 67cb25
      if (r < b && r > a)
Packit 67cb25
        {
Packit 67cb25
          double u = (r - a) / dx;
Packit 67cb25
          double f = modf(u, &bin;;
Packit 67cb25
          j = (int)bin;
Packit 67cb25
Packit 67cb25
          if (f == 0)
Packit 67cb25
            edge[j]++;
Packit 67cb25
          else 
Packit 67cb25
            count[j]++;
Packit 67cb25
        }
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  /* Sort out where the hits on the edges should go */
Packit 67cb25
Packit 67cb25
  for (i = 0; i < BINS; i++)
Packit 67cb25
    {
Packit 67cb25
      /* If the bin above is empty, its lower edge hits belong in the
Packit 67cb25
         lower bin */
Packit 67cb25
Packit 67cb25
      if (i + 1 < BINS && count[i+1] == 0) {
Packit 67cb25
        count[i] += edge[i+1];
Packit 67cb25
        edge[i+1] = 0;
Packit 67cb25
      }
Packit 67cb25
Packit 67cb25
      count[i] += edge[i];
Packit 67cb25
      edge[i] = 0;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  mean = (total / n);
Packit 67cb25
Packit 67cb25
  status = !gsl_finite(mean);
Packit 67cb25
  if (status) {
Packit 67cb25
    gsl_test (status, "%s, finite mean, observed %g", name, mean);
Packit 67cb25
    return;
Packit 67cb25
  }
Packit 67cb25
Packit 67cb25
  for (i = 0; i < BINS; i++)
Packit 67cb25
    {
Packit 67cb25
      double x = a + i * dx;
Packit 67cb25
      double d = fabs (count[i] - n * p[i]);
Packit 67cb25
      if (!gsl_finite(p[i])) 
Packit 67cb25
        {
Packit 67cb25
          status_i = 1;
Packit 67cb25
        }
Packit 67cb25
      else if (p[i] != 0)
Packit 67cb25
        {
Packit 67cb25
          double s = d / sqrt (n * p[i]);
Packit 67cb25
          status_i = (s > 5) && (d > 2);
Packit 67cb25
        }
Packit 67cb25
      else
Packit 67cb25
        {
Packit 67cb25
          status_i = (count[i] != 0);
Packit 67cb25
        }
Packit 67cb25
      
Packit 67cb25
      /* Extend the sample if there is an outlier on the first attempt
Packit 67cb25
         to avoid spurious failures when running large numbers of tests. */
Packit 67cb25
      if (status_i && attempts < 50) 
Packit 67cb25
        { 
Packit 67cb25
          n0 = n; 
Packit 67cb25
          n = 2.0*n;
Packit 67cb25
          goto trial;
Packit 67cb25
        }
Packit 67cb25
Packit 67cb25
      status |= status_i;
Packit 67cb25
      if (status_i)
Packit 67cb25
        gsl_test (status_i, "%s [%g,%g) (%g/%d=%g observed vs %g expected)",
Packit 67cb25
                  name, x, x + dx, count[i], n, count[i] / n, p[i]);
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  if (status == 0)
Packit 67cb25
    gsl_test (status, "%s, sampling against pdf over range [%g,%g) ",
Packit 67cb25
              name, a, b);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
void
Packit 67cb25
testDiscretePDF (double (*f) (void), double (*pdf) (unsigned int),
Packit 67cb25
                 const char *name)
Packit 67cb25
{
Packit 67cb25
  double count[BINS], p[BINS];
Packit 67cb25
  unsigned int i;
Packit 67cb25
  int status = 0, status_i = 0;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < BINS; i++)
Packit 67cb25
    count[i] = 0;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < N; i++)
Packit 67cb25
    {
Packit 67cb25
      int r = (int) (f ());
Packit 67cb25
      if (r >= 0 && r < BINS)
Packit 67cb25
        count[r]++;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  for (i = 0; i < BINS; i++)
Packit 67cb25
    p[i] = pdf (i);
Packit 67cb25
Packit 67cb25
  for (i = 0; i < BINS; i++)
Packit 67cb25
    {
Packit 67cb25
      double d = fabs (count[i] - N * p[i]);
Packit 67cb25
      if (p[i] != 0)
Packit 67cb25
        {
Packit 67cb25
          double s = d / sqrt (N * p[i]);
Packit 67cb25
          status_i = (s > 5) && (d > 1);
Packit 67cb25
        }
Packit 67cb25
      else
Packit 67cb25
        {
Packit 67cb25
          status_i = (count[i] != 0);
Packit 67cb25
        }
Packit 67cb25
      status |= status_i;
Packit 67cb25
      if (status_i)
Packit 67cb25
        gsl_test (status_i, "%s i=%d (%g observed vs %g expected)",
Packit 67cb25
                  name, i, count[i] / N, p[i]);
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  if (status == 0)
Packit 67cb25
    gsl_test (status, "%s, sampling against pdf over range [%d,%d) ",
Packit 67cb25
              name, 0, BINS);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_beta (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_beta (r_global, 2.0, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_beta_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_beta_pdf (x, 2.0, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bernoulli (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_bernoulli (r_global, 0.3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bernoulli_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_bernoulli_pdf (n, 0.3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial (r_global, 0.3, 5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 0.3, 5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial0 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial (r_global, 0, 8);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial0_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 0, 8);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial (r_global, 1, 8);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial1_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 1, 8);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_knuth (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_knuth (r_global, 0.3, 5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_knuth_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 0.3, 5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_large (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial (r_global, 0.3, 55);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_large_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 0.3, 55);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_large_knuth (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_knuth (r_global, 0.3, 55);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_large_knuth_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 0.3, 55);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_huge (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial (r_global, 0.3, 5500);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_huge_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 0.3, 5500);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_huge_knuth (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_knuth (r_global, 0.3, 5500);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_huge_knuth_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 0.3, 5500);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_max (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial (r_global, 1e-8, 1<<31);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_binomial_max_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_binomial_pdf (n, 1e-8, 1<<31);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_cauchy (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_cauchy (r_global, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_cauchy_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_cauchy_pdf (x, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_chisq (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_chisq (r_global, 13.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_chisq_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_chisq_pdf (x, 13.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_chisqnu2 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_chisq (r_global, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_chisqnu2_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_chisq_pdf (x, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir2d (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0, theta;
Packit 67cb25
  gsl_ran_dir_2d (r_global, &x, &y);
Packit 67cb25
  theta = atan2 (x, y);
Packit 67cb25
  return theta;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir2d_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  if (x > -M_PI && x <= M_PI)
Packit 67cb25
    {
Packit 67cb25
      return 1 / (2 * M_PI);
Packit 67cb25
    }
Packit 67cb25
  else
Packit 67cb25
    {
Packit 67cb25
      return 0;
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir2d_trig_method (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0, theta;
Packit 67cb25
  gsl_ran_dir_2d_trig_method (r_global, &x, &y);
Packit 67cb25
  theta = atan2 (x, y);
Packit 67cb25
  return theta;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir2d_trig_method_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  if (x > -M_PI && x <= M_PI)
Packit 67cb25
    {
Packit 67cb25
      return 1 / (2 * M_PI);
Packit 67cb25
    }
Packit 67cb25
  else
Packit 67cb25
    {
Packit 67cb25
      return 0;
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir3dxy (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0, z = 0, theta;
Packit 67cb25
  gsl_ran_dir_3d (r_global, &x, &y, &z);
Packit 67cb25
  theta = atan2 (x, y);
Packit 67cb25
  return theta;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir3dxy_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  if (x > -M_PI && x <= M_PI)
Packit 67cb25
    {
Packit 67cb25
      return 1 / (2 * M_PI);
Packit 67cb25
    }
Packit 67cb25
  else
Packit 67cb25
    {
Packit 67cb25
      return 0;
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir3dyz (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0, z = 0, theta;
Packit 67cb25
  gsl_ran_dir_3d (r_global, &x, &y, &z);
Packit 67cb25
  theta = atan2 (y, z);
Packit 67cb25
  return theta;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir3dyz_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  if (x > -M_PI && x <= M_PI)
Packit 67cb25
    {
Packit 67cb25
      return 1 / (2 * M_PI);
Packit 67cb25
    }
Packit 67cb25
  else
Packit 67cb25
    {
Packit 67cb25
      return 0;
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir3dzx (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0, z = 0, theta;
Packit 67cb25
  gsl_ran_dir_3d (r_global, &x, &y, &z);
Packit 67cb25
  theta = atan2 (z, x);
Packit 67cb25
  return theta;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dir3dzx_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  if (x > -M_PI && x <= M_PI)
Packit 67cb25
    {
Packit 67cb25
      return 1 / (2 * M_PI);
Packit 67cb25
    }
Packit 67cb25
  else
Packit 67cb25
    {
Packit 67cb25
      return 0;
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dirichlet (void)
Packit 67cb25
{
Packit 67cb25
  /* This is a bit of a lame test, since when K=2, the Dirichlet distribution
Packit 67cb25
     becomes a beta distribution */
Packit 67cb25
  size_t K = 2;
Packit 67cb25
  double alpha[2] = { 2.5, 5.0 };
Packit 67cb25
  double theta[2] = { 0.0, 0.0 };
Packit 67cb25
Packit 67cb25
  gsl_ran_dirichlet (r_global, K, alpha, theta);
Packit 67cb25
Packit 67cb25
  return theta[0];
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dirichlet_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  size_t K = 2;
Packit 67cb25
  double alpha[2] = { 2.5, 5.0 };
Packit 67cb25
  double theta[2];
Packit 67cb25
Packit 67cb25
  if (x <= 0.0 || x >= 1.0)
Packit 67cb25
    return 0.0;                 /* Out of range */
Packit 67cb25
Packit 67cb25
  theta[0] = x;
Packit 67cb25
  theta[1] = 1.0 - x;
Packit 67cb25
Packit 67cb25
  return gsl_ran_dirichlet_pdf (K, alpha, theta);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dirichlet_small (void)
Packit 67cb25
{
Packit 67cb25
  size_t K = 2;
Packit 67cb25
  double alpha[2] = { 2.5e-3, 5.0e-3};
Packit 67cb25
  double theta[2] = { 0.0, 0.0 };
Packit 67cb25
Packit 67cb25
  gsl_ran_dirichlet (r_global, K, alpha, theta);
Packit 67cb25
Packit 67cb25
  return theta[0];
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_dirichlet_small_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  size_t K = 2;
Packit 67cb25
  double alpha[2] = { 2.5e-3, 5.0e-3 };
Packit 67cb25
  double theta[2];
Packit 67cb25
Packit 67cb25
  if (x <= 0.0 || x >= 1.0)
Packit 67cb25
    return 0.0;                 /* Out of range */
Packit 67cb25
Packit 67cb25
  theta[0] = x;
Packit 67cb25
  theta[1] = 1.0 - x;
Packit 67cb25
Packit 67cb25
  return gsl_ran_dirichlet_pdf (K, alpha, theta);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
/* Check that the observed means of the Dirichlet variables are
Packit 67cb25
   within reasonable statistical errors of their correct values. */
Packit 67cb25
Packit 67cb25
#define DIRICHLET_K 10
Packit 67cb25
Packit 67cb25
void
Packit 67cb25
test_dirichlet_moments (void)
Packit 67cb25
{
Packit 67cb25
  double alpha[DIRICHLET_K];
Packit 67cb25
  double theta[DIRICHLET_K];
Packit 67cb25
  double theta_sum[DIRICHLET_K];
Packit 67cb25
Packit 67cb25
  double alpha_sum = 0.0;
Packit 67cb25
  double mean, obs_mean, sd, sigma;
Packit 67cb25
  int status, k, n;
Packit 67cb25
Packit 67cb25
  for (k = 0; k < DIRICHLET_K; k++)
Packit 67cb25
    {
Packit 67cb25
      alpha[k] = gsl_ran_exponential (r_global, 0.1);
Packit 67cb25
      alpha_sum += alpha[k];
Packit 67cb25
      theta_sum[k] = 0.0;
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  for (n = 0; n < N; n++)
Packit 67cb25
    {
Packit 67cb25
      gsl_ran_dirichlet (r_global, DIRICHLET_K, alpha, theta);
Packit 67cb25
      for (k = 0; k < DIRICHLET_K; k++)
Packit 67cb25
        theta_sum[k] += theta[k];
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  for (k = 0; k < DIRICHLET_K; k++)
Packit 67cb25
    {
Packit 67cb25
      mean = alpha[k] / alpha_sum;
Packit 67cb25
      sd =
Packit 67cb25
        sqrt ((alpha[k] * (1. - alpha[k] / alpha_sum)) /
Packit 67cb25
              (alpha_sum * (alpha_sum + 1.)));
Packit 67cb25
      obs_mean = theta_sum[k] / N;
Packit 67cb25
      sigma = sqrt ((double) N) * fabs (mean - obs_mean) / sd;
Packit 67cb25
Packit 67cb25
      status = (sigma > 3.0);
Packit 67cb25
Packit 67cb25
      gsl_test (status,
Packit 67cb25
                "test gsl_ran_dirichlet: mean (%g observed vs %g expected)",
Packit 67cb25
                obs_mean, mean);
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
/* Check that the observed means of the multinomial variables are
Packit 67cb25
   within reasonable statistical errors of their correct values. */
Packit 67cb25
Packit 67cb25
void
Packit 67cb25
test_multinomial_moments (void)
Packit 67cb25
{
Packit 67cb25
  const unsigned int sum_n = 100;
Packit 67cb25
Packit 67cb25
  const double p[MULTI_DIM] ={ 0.2, 0.20, 0.17, 0.14, 0.12,
Packit 67cb25
                               0.07, 0.05, 0.02, 0.02, 0.01 };
Packit 67cb25
Packit 67cb25
  unsigned int  x[MULTI_DIM];
Packit 67cb25
  double x_sum[MULTI_DIM];
Packit 67cb25
Packit 67cb25
  double mean, obs_mean, sd, sigma;
Packit 67cb25
  int status, k, n;
Packit 67cb25
Packit 67cb25
  for (k = 0; k < MULTI_DIM; k++)
Packit 67cb25
    x_sum[k] =0.0;
Packit 67cb25
Packit 67cb25
  for (n = 0; n < N; n++)
Packit 67cb25
    {
Packit 67cb25
      gsl_ran_multinomial (r_global, MULTI_DIM, sum_n, p, x);
Packit 67cb25
      for (k = 0; k < MULTI_DIM; k++)
Packit 67cb25
        x_sum[k] += x[k];
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  for (k = 0; k < MULTI_DIM; k++)
Packit 67cb25
    {
Packit 67cb25
      mean = p[k] * sum_n;
Packit 67cb25
      sd = p[k] * (1.-p[k]) * sum_n;
Packit 67cb25
Packit 67cb25
      obs_mean = x_sum[k] / N;
Packit 67cb25
      sigma = sqrt ((double) N) * fabs (mean - obs_mean) / sd;
Packit 67cb25
Packit 67cb25
      status = (sigma > 3.0);
Packit 67cb25
Packit 67cb25
      gsl_test (status,
Packit 67cb25
                "test gsl_ran_multinomial: mean (%g observed vs %g expected)",
Packit 67cb25
                obs_mean, mean);
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_discrete1 (void)
Packit 67cb25
{
Packit 67cb25
  static double P[3] = { 0.59, 0.4, 0.01 };
Packit 67cb25
  if (g1 == NULL)
Packit 67cb25
    {
Packit 67cb25
      g1 = gsl_ran_discrete_preproc (3, P);
Packit 67cb25
    }
Packit 67cb25
  return gsl_ran_discrete (r_global, g1);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_discrete1_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_discrete_pdf ((size_t) n, g1);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_discrete2 (void)
Packit 67cb25
{
Packit 67cb25
  static double P[10] = { 1, 9, 3, 4, 5, 8, 6, 7, 2, 0 };
Packit 67cb25
  if (g2 == NULL)
Packit 67cb25
    {
Packit 67cb25
      g2 = gsl_ran_discrete_preproc (10, P);
Packit 67cb25
    }
Packit 67cb25
  return gsl_ran_discrete (r_global, g2);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_discrete2_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_discrete_pdf ((size_t) n, g2);
Packit 67cb25
}
Packit 67cb25
double
Packit 67cb25
test_discrete3 (void)
Packit 67cb25
{
Packit 67cb25
  static double P[20];
Packit 67cb25
  if (g3 == NULL)
Packit 67cb25
    { int i;
Packit 67cb25
      for (i=0; i<20; ++i) P[i]=1.0/20;
Packit 67cb25
      g3 = gsl_ran_discrete_preproc (20, P);
Packit 67cb25
    }
Packit 67cb25
  return gsl_ran_discrete (r_global, g3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_discrete3_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_discrete_pdf ((size_t) n, g3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_erlang (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_erlang (r_global, 3.0, 4.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_erlang_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_erlang_pdf (x, 3.0, 4.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exponential (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exponential (r_global, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exponential_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exponential_pdf (x, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow0 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow (r_global, 3.7, 0.3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow0_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow_pdf (x, 3.7, 0.3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow (r_global, 3.7, 1.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow_pdf (x, 3.7, 1.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow1a (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow (r_global, 3.7, 1.9);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow1a_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow_pdf (x, 3.7, 1.9);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow2 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow (r_global, 3.7, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow2_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow_pdf (x, 3.7, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow2a (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow (r_global, 3.7, 3.5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow2a_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow_pdf (x, 3.7, 3.5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow2b (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow (r_global, 3.7, 7.5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_exppow2b_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_exppow_pdf (x, 3.7, 7.5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_fdist (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_fdist (r_global, 3.0, 4.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_fdist_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_fdist_pdf (x, 3.0, 4.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
/* Test case for bug #28500: overflow in gsl_ran_fdist_pdf */
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_fdist_large (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_fdist (r_global, 8.0, 249.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_fdist_large_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_fdist_pdf (x, 8.0, 249.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_flat (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_flat (r_global, 3.0, 4.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_flat_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_flat_pdf (x, 3.0, 4.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma (r_global, 2.5, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 2.5, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma (r_global, 1.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 1.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_int (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma (r_global, 10.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_int_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 10.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_large (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma (r_global, 20.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_large_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 20.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_small (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma (r_global, 0.92, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_small_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 0.92, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_vlarge (void)
Packit 67cb25
{
Packit 67cb25
  /* Scale the distribution to get it into the range [-5,5] */
Packit 67cb25
  double c = 2.71828181565;
Packit 67cb25
  double b = 6.32899304917e-10;
Packit 67cb25
  double d = 1e4;
Packit 67cb25
  return (gsl_ran_gamma (r_global, 4294967296.0, b) - c) * d;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_vlarge_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  double c = 2.71828181565;
Packit 67cb25
  double b = 6.32899304917e-10;
Packit 67cb25
  double d = 1e4;
Packit 67cb25
  return gsl_ran_gamma_pdf ((x / d) + c, 4294967296.0, b) / d;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_mt (r_global, 2.5, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 2.5, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_mt (r_global, 1.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 1.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt_int (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_mt (r_global, 10.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt_int_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 10.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt_large (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_mt (r_global, 20.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt_large_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 20.0, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt_small (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_mt (r_global, 0.92, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_mt_small_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gamma_pdf (x, 0.92, 2.17);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_knuth_vlarge (void)
Packit 67cb25
{
Packit 67cb25
  /* Scale the distribution to get it into the range [-5,5] */
Packit 67cb25
  double c = 2.71828181565;
Packit 67cb25
  double b = 6.32899304917e-10;
Packit 67cb25
  double d = 1e4;
Packit 67cb25
  return (gsl_ran_gamma_knuth (r_global, 4294967296.0, b) - c) * d;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gamma_knuth_vlarge_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  double c = 2.71828181565;
Packit 67cb25
  double b = 6.32899304917e-10;
Packit 67cb25
  double d = 1e4;
Packit 67cb25
  return gsl_ran_gamma_pdf ((x / d) + c, 4294967296.0, b) / d;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian (r_global, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_ratio_method (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_ratio_method (r_global, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_ratio_method_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_ziggurat (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_ziggurat (r_global, 3.12);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_ziggurat_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, 3.12);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_tail (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_tail (r_global, 1.7, 0.25);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_tail_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_tail_pdf (x, 1.7, 0.25);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_tail1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_tail (r_global, -1.7, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_tail1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_tail_pdf (x, -1.7, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_tail2 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_tail (r_global, 0.1, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gaussian_tail2_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_tail_pdf (x, 0.1, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_ugaussian (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_ugaussian (r_global);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_ugaussian_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_ugaussian_pdf (x);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_ugaussian_ratio_method (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_ugaussian_ratio_method (r_global);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_ugaussian_ratio_method_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_ugaussian_pdf (x);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_ugaussian_tail (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_ugaussian_tail (r_global, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_ugaussian_tail_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_ugaussian_tail_pdf (x, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bivariate_gaussian1 (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0;
Packit 67cb25
  gsl_ran_bivariate_gaussian (r_global, 3.0, 2.0, 0.3, &x, &y);
Packit 67cb25
  return x;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bivariate_gaussian1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, 3.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bivariate_gaussian2 (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0;
Packit 67cb25
  gsl_ran_bivariate_gaussian (r_global, 3.0, 2.0, 0.3, &x, &y);
Packit 67cb25
  return y;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bivariate_gaussian2_pdf (double y)
Packit 67cb25
{
Packit 67cb25
  int i, n = 10;
Packit 67cb25
  double sum = 0;
Packit 67cb25
  double a = -10, b = 10, dx = (b - a) / n;
Packit 67cb25
  for (i = 0; i < n; i++)
Packit 67cb25
    {
Packit 67cb25
      double x = a + i * dx;
Packit 67cb25
      sum += gsl_ran_bivariate_gaussian_pdf (x, y, 3.0, 2.0, 0.3) * dx;
Packit 67cb25
    }
Packit 67cb25
  return sum;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bivariate_gaussian3 (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0;
Packit 67cb25
  gsl_ran_bivariate_gaussian (r_global, 3.0, 2.0, 0.3, &x, &y);
Packit 67cb25
  return x + y;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bivariate_gaussian3_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  double sx = 3.0, sy = 2.0, r = 0.3;
Packit 67cb25
  double su = (sx + r * sy);
Packit 67cb25
  double sv = sy * sqrt (1 - r * r);
Packit 67cb25
  double sigma = sqrt (su * su + sv * sv);
Packit 67cb25
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, sigma);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bivariate_gaussian4 (void)
Packit 67cb25
{
Packit 67cb25
  double x = 0, y = 0;
Packit 67cb25
  gsl_ran_bivariate_gaussian (r_global, 3.0, 2.0, -0.5, &x, &y);
Packit 67cb25
  return x + y;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_bivariate_gaussian4_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  double sx = 3.0, sy = 2.0, r = -0.5;
Packit 67cb25
  double su = (sx + r * sy);
Packit 67cb25
  double sv = sy * sqrt (1 - r * r);
Packit 67cb25
  double sigma = sqrt (su * su + sv * sv);
Packit 67cb25
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, sigma);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
/* Examples from R (GPL): http://www.r-project.org/
Packit 67cb25
 * library(mvtnorm); packageVersion("mvtnorm") # 1.0.5
Packit 67cb25
 * mu <- c(1, 2)
Packit 67cb25
 * Sigma <- matrix(c(4,2, 2,3), ncol=2)
Packit 67cb25
 * x <- c(0, 0)
Packit 67cb25
 * sprintf("%.15f", dmvnorm(x=x, mean=mu, sigma=Sigma, log=TRUE)) # -3.565097837249263
Packit 67cb25
 */
Packit 67cb25
void
Packit 67cb25
test_multivariate_gaussian_log_pdf (void)
Packit 67cb25
{
Packit 67cb25
  size_t d = 2;
Packit 67cb25
  const double exp_res = -3.565097837249263;
Packit 67cb25
  double obs_res;
Packit 67cb25
  gsl_vector * mu = gsl_vector_calloc(d);
Packit 67cb25
  gsl_matrix * Sigma = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * L = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_vector * x = gsl_vector_calloc(d);
Packit 67cb25
  gsl_vector * work = gsl_vector_calloc(d);
Packit 67cb25
Packit 67cb25
  gsl_vector_set(mu, 0, 1);
Packit 67cb25
  gsl_vector_set(mu, 1, 2);
Packit 67cb25
Packit 67cb25
  gsl_matrix_set(Sigma, 0, 0, 4);
Packit 67cb25
  gsl_matrix_set(Sigma, 1, 1, 3);
Packit 67cb25
  gsl_matrix_set(Sigma, 0, 1, 2);
Packit 67cb25
  gsl_matrix_set(Sigma, 1, 0, 2);
Packit 67cb25
Packit 67cb25
  gsl_matrix_memcpy(L, Sigma);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(L);
Packit 67cb25
Packit 67cb25
  gsl_ran_multivariate_gaussian_log_pdf(x, mu, L, &obs_res, work);
Packit 67cb25
  gsl_test_rel(obs_res, exp_res, 1.0e-10, "gsl_ran_multivariate_gaussian_log_pdf");
Packit 67cb25
Packit 67cb25
  gsl_vector_free(mu);
Packit 67cb25
  gsl_matrix_free(Sigma);
Packit 67cb25
  gsl_matrix_free(L);
Packit 67cb25
  gsl_vector_free(x);
Packit 67cb25
  gsl_vector_free(work);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
/* Examples from R (GPL): http://www.r-project.org/
Packit 67cb25
 * library(mvtnorm); packageVersion("mvtnorm") # 1.0.5
Packit 67cb25
 * mu <- c(1, 2)
Packit 67cb25
 * Sigma <- matrix(c(4,2, 2,3), ncol=2)
Packit 67cb25
 * x <- c(0, 0)
Packit 67cb25
 * sprintf("%.15f", dmvnorm(x=x, mean=mu, sigma=Sigma, log=FALSE)) # 0.028294217120391
Packit 67cb25
 */
Packit 67cb25
void
Packit 67cb25
test_multivariate_gaussian_pdf (void)
Packit 67cb25
{
Packit 67cb25
  size_t d = 2;
Packit 67cb25
  const double exp_res = 0.028294217120391;
Packit 67cb25
  double obs_res = 0;
Packit 67cb25
  gsl_vector * mu = gsl_vector_calloc(d);
Packit 67cb25
  gsl_matrix * Sigma = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * L = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_vector * x = gsl_vector_calloc(d);
Packit 67cb25
  gsl_vector * work = gsl_vector_calloc(d);
Packit 67cb25
Packit 67cb25
  gsl_vector_set(mu, 0, 1);
Packit 67cb25
  gsl_vector_set(mu, 1, 2);
Packit 67cb25
Packit 67cb25
  gsl_matrix_set(Sigma, 0, 0, 4);
Packit 67cb25
  gsl_matrix_set(Sigma, 1, 1, 3);
Packit 67cb25
  gsl_matrix_set(Sigma, 0, 1, 2);
Packit 67cb25
  gsl_matrix_set(Sigma, 1, 0, 2);
Packit 67cb25
Packit 67cb25
  gsl_matrix_memcpy(L, Sigma);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(L);
Packit 67cb25
Packit 67cb25
  gsl_ran_multivariate_gaussian_pdf(x, mu, L, &obs_res, work);
Packit 67cb25
  gsl_test_rel(obs_res, exp_res, 1.0e-10, "gsl_ran_multivariate_gaussian_pdf");
Packit 67cb25
Packit 67cb25
  gsl_vector_free(mu);
Packit 67cb25
  gsl_matrix_free(Sigma);
Packit 67cb25
  gsl_matrix_free(L);
Packit 67cb25
  gsl_vector_free(x);
Packit 67cb25
  gsl_vector_free(work);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
/* Draw N random vectors according to a given MVN(mu,Sigma). Then, check that
Packit 67cb25
 * one can't reject the null hypothesis that the sample mean is equal to
Packit 67cb25
 * the true mean, using Hotelling's test statistic at 95% confidence level.
Packit 67cb25
 * Details in "Applied Multivariate Statistical Analysis" by  Johnson & Wichern
Packit 67cb25
 * (2001), section 5, page 212.
Packit 67cb25
 */
Packit 67cb25
void
Packit 67cb25
test_multivariate_gaussian (void)
Packit 67cb25
{
Packit 67cb25
  size_t d = 2, i = 0;
Packit 67cb25
  int status = 0;
Packit 67cb25
  double T2 = 0, threshold = 0, alpha = 0.05, pvalue = 0;
Packit 67cb25
  gsl_vector * mu = gsl_vector_calloc(d);
Packit 67cb25
  gsl_matrix * Sigma = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * L = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_vector * sample = gsl_vector_calloc(d);
Packit 67cb25
  gsl_matrix * samples = gsl_matrix_calloc(N, d);
Packit 67cb25
  gsl_vector * mu_hat = gsl_vector_calloc(d);
Packit 67cb25
  gsl_matrix * Sigma_hat = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_vector * mu_hat_ctr = gsl_vector_calloc(d);
Packit 67cb25
  gsl_matrix * Sigma_hat_inv = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_vector * tmp = gsl_vector_calloc(d);
Packit 67cb25
Packit 67cb25
  /* set the true values of parameters mu and Sigma */
Packit 67cb25
  gsl_vector_set(mu, 0, 1);
Packit 67cb25
  gsl_vector_set(mu, 1, 2);
Packit 67cb25
  gsl_matrix_set(Sigma, 0, 0, 4);
Packit 67cb25
  gsl_matrix_set(Sigma, 1, 1, 3);
Packit 67cb25
  gsl_matrix_set(Sigma, 0, 1, 2);
Packit 67cb25
  gsl_matrix_set(Sigma, 1, 0, 2);
Packit 67cb25
Packit 67cb25
  /* draw N random vectors */
Packit 67cb25
  gsl_matrix_memcpy(L, Sigma);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(L);
Packit 67cb25
  for (i = 0; i < N; ++i) {
Packit 67cb25
    gsl_ran_multivariate_gaussian(r_global, mu, L, sample);
Packit 67cb25
    gsl_matrix_set_row(samples, i, sample);
Packit 67cb25
  }
Packit 67cb25
Packit 67cb25
  /* compute the maximum-likelihood estimates */
Packit 67cb25
  gsl_ran_multivariate_gaussian_mean (samples, mu_hat);
Packit 67cb25
  gsl_ran_multivariate_gaussian_vcov (samples, Sigma_hat);
Packit 67cb25
Packit 67cb25
  /* compute Hotelling's test statistic:
Packit 67cb25
     T^2 = n (hat{mu} - mu)' hat{Sigma}^-1 (hat{mu} - mu) */
Packit 67cb25
  gsl_vector_memcpy(mu_hat_ctr, mu_hat);
Packit 67cb25
  gsl_vector_sub(mu_hat_ctr, mu);
Packit 67cb25
  gsl_matrix_memcpy(Sigma_hat_inv, Sigma_hat);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(Sigma_hat_inv);
Packit 67cb25
  gsl_linalg_cholesky_invert(Sigma_hat_inv);
Packit 67cb25
  gsl_blas_dgemv(CblasNoTrans, 1, Sigma_hat_inv, mu_hat_ctr, 0, tmp);
Packit 67cb25
  gsl_blas_ddot(mu_hat_ctr, tmp, &T2;;
Packit 67cb25
  T2 *= N;
Packit 67cb25
Packit 67cb25
  /* test if the null hypothesis (hat{mu} = mu) can be rejected 
Packit 67cb25
   at the alpha level*/
Packit 67cb25
  threshold = (N-1) * d / (double)(N-d) * gsl_cdf_fdist_Pinv(1-alpha, d, N-d);
Packit 67cb25
  status = (T2 > threshold);
Packit 67cb25
  gsl_test(status,
Packit 67cb25
           "test gsl_ran_multivariate_gaussian: T2 %f < %f",
Packit 67cb25
           T2, threshold);
Packit 67cb25
  pvalue = gsl_cdf_fdist_Q(T2, d, N-d);
Packit 67cb25
  status = (pvalue < alpha);
Packit 67cb25
  gsl_test(status,
Packit 67cb25
           "test gsl_ran_multivariate_gaussian: p value %f > %f",
Packit 67cb25
           pvalue, alpha);
Packit 67cb25
Packit 67cb25
  gsl_vector_free(mu);
Packit 67cb25
  gsl_matrix_free(Sigma);
Packit 67cb25
  gsl_matrix_free(L);
Packit 67cb25
  gsl_vector_free(sample);
Packit 67cb25
  gsl_matrix_free(samples);
Packit 67cb25
  gsl_vector_free(mu_hat);
Packit 67cb25
  gsl_matrix_free(Sigma_hat);
Packit 67cb25
  gsl_vector_free(mu_hat_ctr);
Packit 67cb25
  gsl_matrix_free(Sigma_hat_inv);
Packit 67cb25
  gsl_vector_free(tmp);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
/* Examples from R (GPL): http://www.r-project.org/
Packit 67cb25
 * R> version$version.string # R version 3.4.1 (2017-06-30)
Packit 67cb25
 * R> library(MCMCpack); packageVersion("MCMCpack") # 1.3.9
Packit 67cb25
 * R> df <- 3
Packit 67cb25
 * R> V <- matrix(data=c(1, 0.3, 0.3, 1), nrow=2, ncol=2)
Packit 67cb25
 * R> X <- matrix(data=c(2.213322, 1.453357, 1.453357, 3.285779), nrow=2, ncol=2)
Packit 67cb25
 * R> sprintf("%.15f", log(dwish(W=X, v=df, S=V))) # -4.931913612377813
Packit 67cb25
 */
Packit 67cb25
void
Packit 67cb25
test_wishart_log_pdf (void)
Packit 67cb25
{
Packit 67cb25
  size_t d = 2;
Packit 67cb25
  const double df = 3, exp_res = -4.931913612377813;
Packit 67cb25
  double obs_res;
Packit 67cb25
  gsl_matrix * V = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * L = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * X = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * L_X = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * work = gsl_matrix_calloc(d, d);
Packit 67cb25
Packit 67cb25
  gsl_matrix_set(V, 0, 0, 1);
Packit 67cb25
  gsl_matrix_set(V, 1, 1, 1);
Packit 67cb25
  gsl_matrix_set(V, 0, 1, 0.3);
Packit 67cb25
  gsl_matrix_set(V, 1, 0, 0.3);
Packit 67cb25
Packit 67cb25
  gsl_matrix_memcpy(L, V);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(L);
Packit 67cb25
Packit 67cb25
  gsl_matrix_set(X, 0, 0, 2.213322);
Packit 67cb25
  gsl_matrix_set(X, 1, 1, 3.285779);
Packit 67cb25
  gsl_matrix_set(X, 0, 1, 1.453357);
Packit 67cb25
  gsl_matrix_set(X, 1, 0, 1.453357);
Packit 67cb25
Packit 67cb25
  gsl_matrix_memcpy(L_X, X);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(L_X);
Packit 67cb25
Packit 67cb25
  gsl_ran_wishart_log_pdf(X, L_X, df, L, &obs_res, work);
Packit 67cb25
  gsl_test_rel(obs_res, exp_res, 1.0e-10, "gsl_ran_wishart_log_pdf");
Packit 67cb25
Packit 67cb25
  gsl_matrix_free(V);
Packit 67cb25
  gsl_matrix_free(L);
Packit 67cb25
  gsl_matrix_free(X);
Packit 67cb25
  gsl_matrix_free(L_X);
Packit 67cb25
  gsl_matrix_free(work);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
/* Examples from R (GPL): http://www.r-project.org/
Packit 67cb25
 * R> version$version.string # R version 3.4.1 (2017-06-30)
Packit 67cb25
 * R> library(MCMCpack); packageVersion("MCMCpack") # 1.3.9
Packit 67cb25
 * R> df <- 3
Packit 67cb25
 * R> V <- matrix(data=c(1, 0.3, 0.3, 1), nrow=2, ncol=2)
Packit 67cb25
 * R> X <- matrix(data=c(2.213322, 1.453357, 1.453357, 3.285779), nrow=2, ncol=2)
Packit 67cb25
 * R> sprintf("%.15f", dwish(W=X, v=df, S=V)) # 0.007212687778224
Packit 67cb25
 */
Packit 67cb25
void
Packit 67cb25
test_wishart_pdf (void)
Packit 67cb25
{
Packit 67cb25
  size_t d = 2;
Packit 67cb25
  const double df = 3, exp_res = 0.007212687778224;
Packit 67cb25
  double obs_res;
Packit 67cb25
  gsl_matrix * V = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * L = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * X = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * L_X = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * work = gsl_matrix_calloc(d, d);
Packit 67cb25
Packit 67cb25
  gsl_matrix_set(V, 0, 0, 1);
Packit 67cb25
  gsl_matrix_set(V, 1, 1, 1);
Packit 67cb25
  gsl_matrix_set(V, 0, 1, 0.3);
Packit 67cb25
  gsl_matrix_set(V, 1, 0, 0.3);
Packit 67cb25
Packit 67cb25
  gsl_matrix_memcpy(L, V);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(L);
Packit 67cb25
Packit 67cb25
  gsl_matrix_set(X, 0, 0, 2.213322);
Packit 67cb25
  gsl_matrix_set(X, 1, 1, 3.285779);
Packit 67cb25
  gsl_matrix_set(X, 0, 1, 1.453357);
Packit 67cb25
  gsl_matrix_set(X, 1, 0, 1.453357);
Packit 67cb25
Packit 67cb25
  gsl_matrix_memcpy(L_X, X);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(L_X);
Packit 67cb25
Packit 67cb25
  gsl_ran_wishart_pdf(X, L_X, df, L, &obs_res, work);
Packit 67cb25
  gsl_test_rel(obs_res, exp_res, 1.0e-10, "gsl_ran_wishart_pdf");
Packit 67cb25
Packit 67cb25
  gsl_matrix_free(V);
Packit 67cb25
  gsl_matrix_free(L);
Packit 67cb25
  gsl_matrix_free(X);
Packit 67cb25
  gsl_matrix_free(L_X);
Packit 67cb25
  gsl_matrix_free(work);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
/* Draw N random "matrices" according to a W_d(df,V) with d=1 and V=1,
Packit 67cb25
 * and check that their mean and variance are close enough to those of
Packit 67cb25
 * a Chi-squared(df), respectively df and 2df.
Packit 67cb25
 */
Packit 67cb25
void
Packit 67cb25
test_wishart (void)
Packit 67cb25
{
Packit 67cb25
  size_t d = 1, i, j;
Packit 67cb25
  double df[5] = {1, 3, 5, 7, 9}, mean_wishart, var_wishart;
Packit 67cb25
  int status;
Packit 67cb25
  gsl_matrix * V = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * L = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * sample_wishart = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_matrix * work = gsl_matrix_calloc(d, d);
Packit 67cb25
  gsl_vector * samples_wishart = gsl_vector_calloc(N);
Packit 67cb25
Packit 67cb25
  gsl_matrix_set(V, 0, 0, 1.0);
Packit 67cb25
  gsl_matrix_memcpy(L, V);
Packit 67cb25
  gsl_linalg_cholesky_decomp1(L);
Packit 67cb25
Packit 67cb25
  for (j = 0; j < 5; ++j) { /* for loop over df */
Packit 67cb25
Packit 67cb25
    /* draw N random variables from W(df,V) */
Packit 67cb25
    for (i = 0; i < N; ++i) {
Packit 67cb25
      status = gsl_ran_wishart(r_global, df[j], L, sample_wishart, work);
Packit 67cb25
      gsl_vector_set(samples_wishart, i, gsl_matrix_get(sample_wishart, 0, 0));
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
    /* compute their mean and variance */
Packit 67cb25
    mean_wishart = gsl_stats_mean(samples_wishart->data,
Packit 67cb25
                                  samples_wishart->stride, N);
Packit 67cb25
    var_wishart = gsl_stats_variance_m(samples_wishart->data,
Packit 67cb25
                                       samples_wishart->stride,
Packit 67cb25
                                       N, mean_wishart);
Packit 67cb25
Packit 67cb25
    /* check */
Packit 67cb25
    gsl_test_rel(mean_wishart, df[j], 1.0e-1, "gsl_ran_wishart, mean");
Packit 67cb25
    gsl_test_rel(var_wishart, 2*df[j], 1.0e-1, "gsl_ran_wishart, var");
Packit 67cb25
Packit 67cb25
  } /* end of for loop over df */
Packit 67cb25
Packit 67cb25
  gsl_matrix_free(V);
Packit 67cb25
  gsl_matrix_free(L);
Packit 67cb25
  gsl_matrix_free(sample_wishart);
Packit 67cb25
  gsl_matrix_free(work);
Packit 67cb25
  gsl_vector_free(samples_wishart);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_geometric (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_geometric (r_global, 0.5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_geometric_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_geometric_pdf (n, 0.5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_geometric1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_geometric (r_global, 1.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_geometric1_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_geometric_pdf (n, 1.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric (r_global, 5, 7, 4);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric1_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric_pdf (n, 5, 7, 4);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric2 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric (r_global, 5, 7, 11);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric2_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric_pdf (n, 5, 7, 11);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric3 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric (r_global, 5, 7, 1);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric3_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric_pdf (n, 5, 7, 1);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric4 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric (r_global, 5, 7, 20);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric4_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric_pdf (n, 5, 7, 20);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric5 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric (r_global, 2, 7, 5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric5_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric_pdf (n, 2, 7, 5);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric6 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric (r_global, 2, 10, 3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_hypergeometric6_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_hypergeometric_pdf (n, 2, 10, 3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gumbel1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gumbel1 (r_global, 3.12, 4.56);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gumbel1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gumbel1_pdf (x, 3.12, 4.56);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gumbel2 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gumbel2 (r_global, 3.12, 4.56);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_gumbel2_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gumbel2_pdf (x, 3.12, 4.56);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_landau (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_landau (r_global);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_landau_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_landau_pdf (x);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy (r_global, 5.0, 1.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_cauchy_pdf (x, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy2 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy (r_global, 5.0, 2.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy2_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, sqrt (2.0) * 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy1a (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy (r_global, 5.0, 1.01);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy1a_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_cauchy_pdf (x, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy2a (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy (r_global, 5.0, 1.99);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy2a_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, sqrt (2.0) * 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy_skew (r_global, 5.0, 1.0, 0.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_cauchy_pdf (x, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew2 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy_skew (r_global, 5.0, 2.0, 0.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew2_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, sqrt (2.0) * 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew1a (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy_skew (r_global, 5.0, 1.01, 0.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew1a_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_cauchy_pdf (x, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew2a (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy_skew (r_global, 5.0, 1.99, 0.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew2a_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, sqrt (2.0) * 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew1b (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy_skew (r_global, 5.0, 1.01, 0.001);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew1b_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_cauchy_pdf (x, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew2b (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_levy_skew (r_global, 5.0, 1.99, 0.001);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_levy_skew2b_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_gaussian_pdf (x, sqrt (2.0) * 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_logistic (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_logistic (r_global, 3.1);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_logistic_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_logistic_pdf (x, 3.1);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_logarithmic (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_logarithmic (r_global, 0.4);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_logarithmic_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_logarithmic_pdf (n, 0.4);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_lognormal (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_lognormal (r_global, 2.7, 1.3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_lognormal_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_lognormal_pdf (x, 2.7, 1.3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_multinomial (void)
Packit 67cb25
{
Packit 67cb25
  const size_t K = 3;
Packit 67cb25
  const unsigned int sum_n = BINS;
Packit 67cb25
  unsigned int n[3];
Packit 67cb25
  /* Test use of weights instead of probabilities. */
Packit 67cb25
  const double p[] = { 2., 7., 1.};
Packit 67cb25
Packit 67cb25
  gsl_ran_multinomial ( r_global, K, sum_n, p, n);
Packit 67cb25
Packit 67cb25
  return n[0];
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_multinomial_pdf (unsigned int n_0)
Packit 67cb25
{
Packit 67cb25
  /* The margional distribution of just 1 variate  is binomial. */
Packit 67cb25
  size_t K = 2;
Packit 67cb25
  /* Test use of weights instead of probabilities */
Packit 67cb25
  double p[] = { 0.4, 1.6};
Packit 67cb25
  const unsigned int sum_n = BINS;
Packit 67cb25
  unsigned int n[2];
Packit 67cb25
Packit 67cb25
  n[0] = n_0;
Packit 67cb25
  n[1] =sum_n - n_0;
Packit 67cb25
Packit 67cb25
  return gsl_ran_multinomial_pdf (K, p, n);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_multinomial_large (void)
Packit 67cb25
{
Packit 67cb25
  const unsigned int sum_n = BINS;
Packit 67cb25
  unsigned int n[MULTI_DIM];
Packit 67cb25
  const double p[MULTI_DIM] = { 0.2, 0.20, 0.17, 0.14, 0.12,
Packit 67cb25
                                0.07, 0.05, 0.04, 0.01, 0.00  };
Packit 67cb25
Packit 67cb25
  gsl_ran_multinomial ( r_global, MULTI_DIM, sum_n, p, n);
Packit 67cb25
Packit 67cb25
  return n[0];
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_multinomial_large_pdf (unsigned int n_0)
Packit 67cb25
{
Packit 67cb25
  return test_multinomial_pdf(n_0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_negative_binomial (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_negative_binomial (r_global, 0.3, 20.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_negative_binomial_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_negative_binomial_pdf (n, 0.3, 20.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_pascal (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_pascal (r_global, 0.8, 3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_pascal_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_pascal_pdf (n, 0.8, 3);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_pareto (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_pareto (r_global, 1.9, 2.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_pareto_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_pareto_pdf (x, 1.9, 2.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_rayleigh (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_rayleigh (r_global, 1.9);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_rayleigh_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_rayleigh_pdf (x, 1.9);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_rayleigh_tail (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_rayleigh_tail (r_global, 2.7, 1.9);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_rayleigh_tail_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_rayleigh_tail_pdf (x, 2.7, 1.9);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_poisson (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_poisson (r_global, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_poisson_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_poisson_pdf (n, 5.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_poisson_large (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_poisson (r_global, 30.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_poisson_large_pdf (unsigned int n)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_poisson_pdf (n, 30.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_tdist1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_tdist (r_global, 1.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_tdist1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_tdist_pdf (x, 1.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_tdist2 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_tdist (r_global, 12.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_tdist2_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_tdist_pdf (x, 12.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_laplace (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_laplace (r_global, 2.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_laplace_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_laplace_pdf (x, 2.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_weibull (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_weibull (r_global, 3.14, 2.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_weibull_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_weibull_pdf (x, 3.14, 2.75);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_weibull1 (void)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_weibull (r_global, 2.97, 1.0);
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
double
Packit 67cb25
test_weibull1_pdf (double x)
Packit 67cb25
{
Packit 67cb25
  return gsl_ran_weibull_pdf (x, 2.97, 1.0);
Packit 67cb25
}