gtestGAtmoFlux.cxx File Reference

#include <cstdio>
#include "Tools/Flux/GBGLRSAtmoFlux.h"
#include "Tools/Flux/GAtmoFlux.h"
#include "Framework/Conventions/Units.h"
#include <stdlib.h>
#include "TH3D.h"
#include "Framework/Messenger/Messenger.h"

Go to the source code of this file.

Classes
struct	tests

Macros
#define	UNUSED(V)   ((void) V)
	Program used for testing the GAtmoFlux class. More...
#define	LEN(x)   ((sizeof(x)/sizeof(0[x]))/((size_t)(!(sizeof(x) % sizeof(0[x])))))

Typedefs
typedef int	testFunction(char *err)

Functions
int	isclose (double a, double b, double rel_tol, double abs_tol)
int	testGetTotalFlux (char *err)
int	testGetTotalFluxInEnergyRange (char *err)
int	main (int argc, char **argv)

Variables
struct tests	tests []

Macro Definition Documentation

#define LEN

(

x

)

((sizeof(x)/sizeof(0[x]))/((size_t)(!(sizeof(x) % sizeof(0[x])))))

Definition at line 32 of file gtestGAtmoFlux.cxx.

#define UNUSED

(

V

)

((void) V)

Program used for testing the GAtmoFlux class.

gtestGAtmoFlux

Author

Anthony LaTorre <tlatorre at="" uchicago="" dot="" edu>="">

March 31, 2020

Copyright (c) 2003-2020, The GENIE Collaboration For the full text of the license visit http://copyright.genie-mc.org

Definition at line 27 of file gtestGAtmoFlux.cxx.

Typedef Documentation

typedef int testFunction(char *err)

Definition at line 34 of file gtestGAtmoFlux.cxx.

Function Documentation

int isclose	(	double	a,
		double	b,
		double	rel_tol,
		double	abs_tol
	)

Definition at line 36 of file gtestGAtmoFlux.cxx.

{
    /* Returns 1 if a and b are "close". This algorithm is taken from Python's
     * math.isclose() function.
     *
     * See https://www.python.org/dev/peps/pep-0485/. */
    return fabs(a-b) <= fmax(rel_tol*fmax(fabs(a),fabs(b)),abs_tol);
}

int main	(	int	argc,
		char **	argv
	)

Definition at line 201 of file gtestGAtmoFlux.cxx.

{
  unsigned int i;
  char err[256];
  int retval = 0;
  struct tests test;

  UNUSED(argc);
  UNUSED(argv);

  /* Don't print low level messages so we get a nice output. */
  Messenger * msg = Messenger::Instance();
  msg->SetPriorityLevel("Flux", pFATAL);

  for (i = 0; i < LEN(tests); i++) {
    test = tests[i];

    if (!test.test(err)) {
      printf("[\033[92mok\033[0m] %s\n", test.name);
    } else {
      printf("[\033[91mfail\033[0m] %s: %s\n", test.name, err);
      retval = 1;
    }
  }

  return retval;
}

int testGetTotalFlux

(

char *

err

)

Definition at line 49 of file gtestGAtmoFlux.cxx.

{
  GAtmoFlux *atmo_flux_driver;
  double emin, emax, value, expected;
  char filename[256];

  sprintf(filename, "%s/src/contrib/test/fmax20_i0403z.sno_nue", getenv("GENIE"));

  GBGLRSAtmoFlux *bartol_flux = new GBGLRSAtmoFlux;
  atmo_flux_driver = dynamic_cast<GAtmoFlux *>(bartol_flux);

  emin = -1;
  emax = 1e9;

  // Configure GAtmoFlux options (common to all concrete atmospheric flux drivers)
  // set min/max energy:
  atmo_flux_driver->ForceMinEnergy(emin * units::GeV);
  atmo_flux_driver->ForceMaxEnergy(emax * units::GeV);
  // set flux files:
  atmo_flux_driver->AddFluxFile(12, filename);
  atmo_flux_driver->LoadFluxData();

  value = atmo_flux_driver->GetTotalFlux();
  expected = atmo_flux_driver->GetFlux(12);

  /* Test that GetTotalFlux() is the same as GetFlux(12) since we are only
   * including a single neutrino flavour. */
  if (value != expected) {
    sprintf(err, "GetTotalFlux() = %f which is not equal to GetFlux(12) = %f", value, expected);
    goto err;
  }

  delete atmo_flux_driver;

  return 0;

err:
  delete atmo_flux_driver;

  return 1;
}

int testGetTotalFluxInEnergyRange

(

char *

err

)

Definition at line 92 of file gtestGAtmoFlux.cxx.

{
  GAtmoFlux *atmo_flux_driver;
  double emin, emax, value, expected;
  char filename[256];

  sprintf(filename, "%s/src/contrib/test/fmax20_i0403z.sno_nue", getenv("GENIE"));

  GBGLRSAtmoFlux *bartol_flux = new GBGLRSAtmoFlux;
  atmo_flux_driver = dynamic_cast<GAtmoFlux *>(bartol_flux);

  // set flux files:
  atmo_flux_driver->AddFluxFile(12, filename);
  atmo_flux_driver->LoadFluxData();

  emin = 0;
  emax = 1e4;

  // Configure GAtmoFlux options (common to all concrete atmospheric flux drivers)
  // set min/max energy:
  atmo_flux_driver->ForceMinEnergy(emin * units::GeV);
  atmo_flux_driver->ForceMaxEnergy(emax * units::GeV);

  value = atmo_flux_driver->GetTotalFlux();
  expected = atmo_flux_driver->GetTotalFluxInEnergyRange();

  if (value != expected) {
    sprintf(err, "GetTotalFlux(%.2f,%.2f) = %f which is not equal to the expected total flux = %f", emin, emax, value, expected);
    goto err;
  }

  /* Now set emin and emax both to 10 GeV and make sure we get 0. */
  emin = 1e4;
  emax = 1e4;

  atmo_flux_driver->ForceMinEnergy(emin * units::GeV);
  atmo_flux_driver->ForceMaxEnergy(emax * units::GeV);

  value = atmo_flux_driver->GetTotalFluxInEnergyRange();
  expected = 0;

  if (value != expected) {
    sprintf(err, "GetTotalFlux(%.1e,%.1e) = %f, but expected %f!", emin, emax, value, expected);
    goto err;
  }

  /* Now set emin and emax both below the bounds and make sure we get 0. */
  emin = 0;
  emax = 0.01;

  atmo_flux_driver->ForceMinEnergy(emin * units::GeV);
  atmo_flux_driver->ForceMaxEnergy(emax * units::GeV);

  value = atmo_flux_driver->GetTotalFluxInEnergyRange();
  expected = 0;

  if (value != expected) {
    sprintf(err, "GetTotalFlux(%.1e,%.1e) = %f, but expected %f!", emin, emax, value, expected);
    return 1;
  }

  /* Now we test when both emin and emax are in the same bin. */
  emin = 0.106;
  emax = 0.11;

  atmo_flux_driver->ForceMinEnergy(emin * units::GeV);
  atmo_flux_driver->ForceMaxEnergy(emax * units::GeV);

  value = atmo_flux_driver->GetTotalFluxInEnergyRange();
  expected = atmo_flux_driver->GetFlux(12,emin)*(emax-emin);

  if (!isclose(value,expected,1e-5,0)) {
    sprintf(err, "GetTotalFlux(%.3f,%.3f) = %f, but expected %f!", emin, emax, value, expected);
    goto err;
  }

  /* Now we test when emin and emax are just past the low and high bin edges. */
  emin = 0.10 + 1e-10;
  emax = 10.0 - 1e-10;

  atmo_flux_driver->ForceMinEnergy(emin * units::GeV);
  atmo_flux_driver->ForceMaxEnergy(emax * units::GeV);

  value = atmo_flux_driver->GetTotalFluxInEnergyRange();
  expected = atmo_flux_driver->GetTotalFlux();

  if (!isclose(value,expected,1e-5,0)) {
  sprintf(err, "GetTotalFlux(%.3f,%.3f) = %f, but expected %f!", emin, emax, value, expected);
    goto err;
  }

  delete atmo_flux_driver;

  return 0;

err:
  delete atmo_flux_driver;

  return 1;
}

Variable Documentation

struct tests tests[]

Initial value:

= {
    {testGetTotalFlux, "testGetTotalFlux"},
    {testGetTotalFluxInEnergyRange, "testGetTotalFluxInEnergyRange"},
}