#include "histogram.h" 
 
#include <library/cpp/testing/unittest/registar.h>
 
using namespace NMonitoring; 
 
Y_UNIT_TEST_SUITE(THistorgamTest) {
    Y_UNIT_TEST(TakeSnapshot) {
        THdrHistogram h(1, 10, 3); 
        UNIT_ASSERT(h.RecordValue(1)); 
        UNIT_ASSERT(h.RecordValue(2)); 
        UNIT_ASSERT(h.RecordValues(3, 10)); 
        UNIT_ASSERT(h.RecordValue(4)); 
        UNIT_ASSERT(h.RecordValue(5)); 
 
        UNIT_ASSERT_EQUAL(h.GetTotalCount(), 14); 
 
        THistogramSnapshot snapshot; 
        h.TakeSnaphot(&snapshot); 
 
        UNIT_ASSERT_EQUAL(h.GetTotalCount(), 0); 
 
        UNIT_ASSERT_EQUAL(snapshot.Min, 1); 
        UNIT_ASSERT_EQUAL(snapshot.Max, 5); 
 
        // >>> a = [1, 2] + [3 for i in range(10)]  + [4, 5] 
        // >>> numpy.mean(a) 
        // 3.0 
        UNIT_ASSERT_DOUBLES_EQUAL(snapshot.Mean, 3.0, 1e-6); 
 
        // >>> numpy.std(a) 
        // 0.84515425472851657 
        UNIT_ASSERT_DOUBLES_EQUAL(snapshot.StdDeviation, 0.84515425472851657, 1e-6); 
 
        // >>> [(p, round(numpy.percentile(a, p))) for p in [50, 75, 90, 95, 98, 99, 99.9, 100]] 
        // [(50, 3.0), (75, 3.0), (90, 4.0), (95, 4.0), (98, 5.0), (99, 5.0), (99.9, 5.0), (100, 5.0)] 
        UNIT_ASSERT_EQUAL(snapshot.Percentile50, 3); 
        UNIT_ASSERT_EQUAL(snapshot.Percentile75, 3); 
        UNIT_ASSERT_EQUAL(snapshot.Percentile90, 4); 
        UNIT_ASSERT_EQUAL(snapshot.Percentile95, 4); 
        UNIT_ASSERT_EQUAL(snapshot.Percentile98, 5); 
        UNIT_ASSERT_EQUAL(snapshot.Percentile99, 5); 
        UNIT_ASSERT_EQUAL(snapshot.Percentile999, 5); 

        UNIT_ASSERT_EQUAL(snapshot.TotalCount, 14);
    } 
}