Blame movstat/test.c

Packit 67cb25
/* movstat/test.c
Packit 67cb25
 * 
Packit 67cb25
 * Copyright (C) 2018 Patrick Alken
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 <stdlib.h>
Packit 67cb25
#include <math.h>
Packit 67cb25
#include <string.h>
Packit 67cb25
#include <gsl/gsl_math.h>
Packit 67cb25
#include <gsl/gsl_movstat.h>
Packit 67cb25
#include <gsl/gsl_statistics.h>
Packit 67cb25
#include <gsl/gsl_test.h>
Packit 67cb25
#include <gsl/gsl_rng.h>
Packit 67cb25
#include <gsl/gsl_randist.h>
Packit 67cb25
#include <gsl/gsl_ieee_utils.h>
Packit 67cb25
Packit 67cb25
/* compare two vectors */
Packit 67cb25
static void
Packit 67cb25
compare_vectors(const double tol, const gsl_vector * v, const gsl_vector * expected,
Packit 67cb25
                const char * desc)
Packit 67cb25
{
Packit 67cb25
  const size_t n = v->size;
Packit 67cb25
  size_t i;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < n; ++i)
Packit 67cb25
    {
Packit 67cb25
      double vi = gsl_vector_get(v, i);
Packit 67cb25
      double ui = gsl_vector_get(expected, i);
Packit 67cb25
Packit 67cb25
      gsl_test_rel(vi, ui, tol, "%s i=%zu", desc, i);
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
/* generate random vector with elements in [-1,1] */
Packit 67cb25
static void
Packit 67cb25
random_vector(gsl_vector * v, gsl_rng * r)
Packit 67cb25
{
Packit 67cb25
  size_t i;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < v->size; ++i)
Packit 67cb25
    {
Packit 67cb25
      double vi = 2.0 * gsl_rng_uniform(r) - 1.0; /* in [-1,1] */
Packit 67cb25
      gsl_vector_set(v, i, vi);
Packit 67cb25
    }
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
static int
Packit 67cb25
test_noisy_sine(const double sigma, gsl_vector * x, gsl_rng * r)
Packit 67cb25
{
Packit 67cb25
  const size_t n = x->size;
Packit 67cb25
  size_t i;
Packit 67cb25
Packit 67cb25
  for (i = 0; i < n; ++i)
Packit 67cb25
    {
Packit 67cb25
      double ti = (double) i / (n - 1.0);
Packit 67cb25
      double xi = sin(2.0 * M_PI * ti) +
Packit 67cb25
                  sin(2.0 * M_PI * 40.0 * ti);
Packit 67cb25
      double ei = gsl_ran_gaussian(r, sigma);
Packit 67cb25
Packit 67cb25
      gsl_vector_set(x, i, xi + ei);
Packit 67cb25
    }
Packit 67cb25
Packit 67cb25
  return GSL_SUCCESS;
Packit 67cb25
}
Packit 67cb25
Packit 67cb25
#include "test_mad.c"
Packit 67cb25
#include "test_mean.c"
Packit 67cb25
#include "test_median.c"
Packit 67cb25
#include "test_minmax.c"
Packit 67cb25
#include "test_Qn.c"
Packit 67cb25
#include "test_qqr.c"
Packit 67cb25
#include "test_sum.c"
Packit 67cb25
#include "test_Sn.c"
Packit 67cb25
#include "test_variance.c"
Packit 67cb25
Packit 67cb25
int
Packit 67cb25
main()
Packit 67cb25
{
Packit 67cb25
  gsl_rng *r = gsl_rng_alloc(gsl_rng_default);
Packit 67cb25
Packit 67cb25
  test_mean(r);
Packit 67cb25
  test_median(r);
Packit 67cb25
  test_minmax(r);
Packit 67cb25
  test_mad(r);
Packit 67cb25
  test_Qn(r);
Packit 67cb25
  test_qqr(r);
Packit 67cb25
  test_sum(r);
Packit 67cb25
  test_Sn(r);
Packit 67cb25
  test_variance(r);
Packit 67cb25
Packit 67cb25
  gsl_rng_free(r);
Packit 67cb25
Packit 67cb25
  exit (gsl_test_summary());
Packit 67cb25
}