2021-12-04 16:42:11 +00:00

60 lines
1.5 KiB
Go

package histogram
import (
"math"
"testing"
)
func TestFastUnderflow(t *testing.T) {
f := GetFast()
defer PutFast(f)
q := f.Quantile(0.5)
if !math.IsNaN(q) {
t.Fatalf("unexpected quantile for empty histogram; got %v; want %v", q, nan)
}
for i := 0; i < maxSamples; i++ {
f.Update(float64(i))
}
qs := f.Quantiles(nil, []float64{0, 0.5, 1})
if qs[0] != 0 {
t.Fatalf("unexpected quantile value for phi=0; got %v; want %v", qs[0], 0)
}
if qs[1] != maxSamples/2 {
t.Fatalf("unexpected quantile value for phi=0.5; got %v; want %v", qs[1], maxSamples/2)
}
if qs[2] != maxSamples-1 {
t.Fatalf("unexpected quantile value for phi=1; got %v; want %v", qs[2], maxSamples-1)
}
}
func TestFastOverflow(t *testing.T) {
f := GetFast()
defer PutFast(f)
for i := 0; i < maxSamples*10; i++ {
f.Update(float64(i))
}
qs := f.Quantiles(nil, []float64{0, 0.5, 0.9999, 1})
if qs[0] != 0 {
t.Fatalf("unexpected quantile value for phi=0; got %v; want %v", qs[0], 0)
}
median := float64(maxSamples*10-1) / 2
if qs[1] < median*0.9 || qs[1] > median*1.1 {
t.Fatalf("unexpected quantile value for phi=0.5; got %v; want %v", qs[1], median)
}
if qs[2] < maxSamples*10*0.9 {
t.Fatalf("unexpected quantile value for phi=0.9999; got %v; want %v", qs[2], maxSamples*10*0.9)
}
if qs[3] != maxSamples*10-1 {
t.Fatalf("unexpected quantile value for phi=1; got %v; want %v", qs[3], maxSamples*10-1)
}
q := f.Quantile(nan)
if !math.IsNaN(q) {
t.Fatalf("unexpected value for phi=NaN; got %v; want %v", q, nan)
}
}