104 lines
2.5 KiB
C++
104 lines
2.5 KiB
C++
#ifdef WITH_TESTS
|
|
|
|
#include "../Tests.h"
|
|
#include "../../math/Distributions.h"
|
|
#include "../../misc/Time.h"
|
|
|
|
template <typename Dist> void benchDist(Dist dist) {
|
|
|
|
const int cnt = 1024*1024*32;
|
|
float f = 0;
|
|
|
|
auto tick = Time::tick();
|
|
for (int i = 0; i < cnt; ++i) {
|
|
f = dist.getProbability(f);
|
|
}
|
|
auto tock = Time::tick();
|
|
|
|
std::cout.imbue(std::locale("en_US.UTF-8"));
|
|
std::cout << (cnt / (Time::diffMS(tick, tock)+1) * 1000) << "/sec" << std::endl;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST(Distribution, Exp) {
|
|
|
|
ASSERT_NEAR(1.5, Distribution::Exponential<float>::getProbability(1.5, 0.0), 0.001);
|
|
ASSERT_NEAR(1.0, Distribution::Exponential<float>::getProbability(1.0, 0.0), 0.001);
|
|
ASSERT_NEAR(0.5, Distribution::Exponential<float>::getProbability(0.5, 0.0), 0.001);
|
|
|
|
ASSERT_NEAR(0.35, Distribution::Exponential<float>::getProbability(1.5, 1.0), 0.05);
|
|
ASSERT_NEAR(0.35, Distribution::Exponential<float>::getProbability(1.0, 1.0), 0.05);
|
|
ASSERT_NEAR(0.35, Distribution::Exponential<float>::getProbability(0.5, 1.0), 0.05);
|
|
|
|
}
|
|
|
|
TEST(Distribution, Exp_Performance) {
|
|
|
|
Distribution::Exponential<float> dist(0.5);
|
|
benchDist(dist);
|
|
|
|
}
|
|
|
|
TEST(Distribution, BesselHelper) {
|
|
|
|
Distribution::Bessel<float> bessel;
|
|
|
|
const float delta = 0.001;
|
|
ASSERT_NEAR( 1.0, bessel.getModified(0, 0), delta);
|
|
ASSERT_NEAR(11.3, bessel.getModified(0, +4), 0.1);
|
|
ASSERT_NEAR(11.3, bessel.getModified(0, -4), 0.1);
|
|
|
|
}
|
|
|
|
TEST(Distribution, VonMises) {
|
|
|
|
Distribution::VonMises<float> dist4(0, 4);
|
|
Distribution::VonMises<float> dist2(0, 2);
|
|
Distribution::VonMises<float> dist1(0, 1);
|
|
Distribution::VonMises<float> dist05(0, 0.5);
|
|
|
|
const float delta = 0.01;
|
|
ASSERT_NEAR(0.768f, dist4.getProbability(0.0f), delta);
|
|
ASSERT_NEAR(0.515f, dist2.getProbability(0.0f), delta);
|
|
ASSERT_NEAR(0.342f, dist1.getProbability(0.0f), delta);
|
|
ASSERT_NEAR(0.24f, dist05.getProbability(0.0f), delta);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
//#include <fstream>
|
|
//TEST(Distribution, VonMisesDump) {
|
|
|
|
// Distribution::VonMises<float> dist1(0, 4.0);
|
|
// auto dist2 = dist1.getLUT();
|
|
|
|
// std::ofstream f1("/tmp/1.dat");
|
|
// std::ofstream f2("/tmp/2.dat");
|
|
// for (float i = -4; i <= +4; i += 0.001) {
|
|
// f1 << i << " " << dist1.getProbability(i) << std::endl;
|
|
// f2 << i << " " << dist2.getProbability(i) << std::endl;
|
|
// }
|
|
// f1.close();
|
|
// f2.close();
|
|
|
|
//}
|
|
|
|
TEST(Distribution, VonMises_Performance) {
|
|
Distribution::VonMises<float> dist(0, 2.0);
|
|
benchDist(dist);
|
|
}
|
|
|
|
TEST(Distribution, VonMisesLUT_Performance) {
|
|
Distribution::VonMises<float> src(0, 2.0);
|
|
auto dist = src.getLUT();
|
|
benchDist(dist);
|
|
}
|
|
|
|
#endif
|