test_amiseOptimalBandwidth.cc
No OneTemporary
Actions

Size

1 KB

Referenced Files

None

Subscribers

None

test_amiseOptimalBandwidth.cc
View Options

	#include "UnitTest++.h"

	#include "npstat/stat/amiseOptimalBandwidth.hh"
	#include "npstat/stat/Distributions1D.hh"
	#include "npstat/nm/ArrayND.hh"

	using namespace npstat;
	using namespace std;

	namespace {
	class GaussFunctor
	{
	public:
	GaussFunctor(double sigma, double nsigmas, unsigned npoints)
	: g_(0.0, sigma), min_(-sigmansigmas), h_(fabs(2min_/npoints)) {}
	inline double step() const {return h_;}
	double operator()(const unsigned* loc, unsigned n) const
	{
	return g_.density(min_ + (loc + 0.5)h_);
	}

	private:
	GaussFunctor();

	Gauss1D g_;
	double min_;
	double h_;
	};

	// Reproduce the "rule of thumb"
	TEST(amiseOptimalBandwidth)
	{
	const double sigma = 2.5;
	const unsigned nbins = 1000;
	GaussFunctor gf(sigma, 6, nbins);
	ArrayND<double> arr(makeShape(nbins));
	arr.functorFill(gf);

	const double bw = amiseOptimalBwGauss(
	0, 1, const_cast<double*>(arr.data()), nbins, gf.step());
	CHECK_CLOSE(1.0592238410488*sigma, bw, 1.e-4);
	}
	}