TestHistogram.cpp

Go to the documentation of this file.

 
#include <gtest/gtest.h>
#include <cmath>
#include <vector>
#include "Utilities/GSLHistogram.h"
 
class HistogramTest : public ::testing::Test {
protected:
    void SetUp() override {
        // Test setup
    }
};
 
TEST_F(HistogramTest, UniformRanges) {
    gsl_histogram* h = gsl_histogram_alloc(10);
    gsl_histogram_set_ranges_uniform(h, 0.0, 10.0);
 
    EXPECT_EQ(gsl_histogram_bins(h), 10);
 
    // Check range boundaries
    double lower, upper;
    gsl_histogram_get_range(h, 0, &lower, &upper);
    EXPECT_DOUBLE_EQ(lower, 0.0);
    EXPECT_DOUBLE_EQ(upper, 1.0);
 
    gsl_histogram_get_range(h, 9, &lower, &upper);
    EXPECT_DOUBLE_EQ(lower, 9.0);
    EXPECT_DOUBLE_EQ(upper, 10.0);
 
    gsl_histogram_free(h);
}
 
TEST_F(HistogramTest, IncrementAndGet) {
    gsl_histogram* h = gsl_histogram_alloc(10);
    gsl_histogram_set_ranges_uniform(h, 0.0, 10.0);
 
    // Increment some values
    gsl_histogram_increment(h, 1.5);
    gsl_histogram_increment(h, 2.5);
    gsl_histogram_increment(h, 1.7);
 
    EXPECT_DOUBLE_EQ(gsl_histogram_get(h, 1), 2.0);  // 1.5 and 1.7 in bin 1
    EXPECT_DOUBLE_EQ(gsl_histogram_get(h, 2), 1.0);  // 2.5 in bin 2
 
    gsl_histogram_free(h);
}
 
TEST_F(HistogramTest, CustomRanges) {
    gsl_histogram* h           = gsl_histogram_alloc(5);
    std::vector<double> ranges = {0.0, 1.0, 2.0, 5.0, 10.0, 20.0};
    gsl_histogram_set_ranges(h, ranges.data(), ranges.size());
 
    gsl_histogram_increment(h, 1.5);
    gsl_histogram_increment(h, 3.0);
    gsl_histogram_increment(h, 7.5);
 
    EXPECT_DOUBLE_EQ(gsl_histogram_get(h, 1), 1.0);  // 1.5 in bin 1
    EXPECT_DOUBLE_EQ(gsl_histogram_get(h, 2), 1.0);  // 3.0 in bin 2
    EXPECT_DOUBLE_EQ(gsl_histogram_get(h, 3), 1.0);  // 7.5 in bin 3
 
    gsl_histogram_free(h);
}
 
TEST_F(HistogramTest, OutOfRange) {
    gsl_histogram* h = gsl_histogram_alloc(10);
    gsl_histogram_set_ranges_uniform(h, 0.0, 10.0);
 
    // Values outside range should not increment any bin
    gsl_histogram_increment(h, -1.0);
    gsl_histogram_increment(h, 11.0);
 
    // All bins should be zero
    for (size_t i = 0; i < 10; ++i) {
        EXPECT_DOUBLE_EQ(gsl_histogram_get(h, i), 0.0);
    }
 
    gsl_histogram_free(h);
}
 
TEST_F(HistogramTest, Histogram2D) {
    gsl_histogram2d* h = gsl_histogram2d_alloc(5, 5);
    gsl_histogram2d_set_ranges_uniform(h, 0.0, 5.0, 0.0, 5.0);
 
    EXPECT_EQ(gsl_histogram2d_nx(h), 5);
    EXPECT_EQ(gsl_histogram2d_ny(h), 5);
 
    // Increment some values
    gsl_histogram2d_increment(h, 1.5, 2.5);
    gsl_histogram2d_increment(h, 1.7, 2.3);
 
    EXPECT_DOUBLE_EQ(gsl_histogram2d_get(h, 1, 2), 2.0);
 
    gsl_histogram2d_free(h);
}
 
TEST_F(HistogramTest, Histogram2DCustomRanges) {
    gsl_histogram2d* h = gsl_histogram2d_alloc(3, 3);
 
    std::vector<double> xranges = {0.0, 1.0, 3.0, 6.0};
    std::vector<double> yranges = {0.0, 2.0, 4.0, 8.0};
 
    gsl_histogram2d_set_ranges(h, xranges.data(), xranges.size(), yranges.data(), yranges.size());
 
    gsl_histogram2d_increment(h, 2.0, 3.0);
    gsl_histogram2d_increment(h, 4.0, 5.0);
 
    EXPECT_DOUBLE_EQ(gsl_histogram2d_get(h, 1, 1), 1.0);
    EXPECT_DOUBLE_EQ(gsl_histogram2d_get(h, 2, 2), 1.0);
 
    gsl_histogram2d_free(h);
}
 
TEST_F(HistogramTest, Histogram2DOutOfRange) {
    gsl_histogram2d* h = gsl_histogram2d_alloc(5, 5);
    gsl_histogram2d_set_ranges_uniform(h, 0.0, 5.0, 0.0, 5.0);
 
    // Values outside range
    gsl_histogram2d_increment(h, -1.0, 2.0);
    gsl_histogram2d_increment(h, 2.0, -1.0);
    gsl_histogram2d_increment(h, 6.0, 2.0);
    gsl_histogram2d_increment(h, 2.0, 6.0);
 
    // All bins should be zero
    for (size_t i = 0; i < 5; ++i) {
        for (size_t j = 0; j < 5; ++j) {
            EXPECT_DOUBLE_EQ(gsl_histogram2d_get(h, i, j), 0.0);
        }
    }
 
    gsl_histogram2d_free(h);
}