mirror of
https://github.com/prometheus/prometheus.git
synced 2024-11-10 07:34:04 -08:00
2208 lines
62 KiB
Go
2208 lines
62 KiB
Go
// Copyright 2021 The Prometheus Authors
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package histogram
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFloatHistogramMul(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
in *FloatHistogram
|
|
scale float64
|
|
expected *FloatHistogram
|
|
}{
|
|
{
|
|
"zero value",
|
|
&FloatHistogram{},
|
|
3.1415,
|
|
&FloatHistogram{},
|
|
},
|
|
{
|
|
"zero multiplier",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 0,
|
|
Count: 0,
|
|
Sum: 0,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{0, 0, 0, 0},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{0, 0, 0, 0},
|
|
},
|
|
},
|
|
{
|
|
"no-op",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
1,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
},
|
|
{
|
|
"double",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
2,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 3493.3 * 2,
|
|
Sum: 2349209.324 * 2,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{2, 6.6, 8.4, 0.2},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{6.2, 6, 1.234e5 * 2, 2000},
|
|
},
|
|
},
|
|
{
|
|
"triple",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 23,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
3,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 33,
|
|
Count: 90,
|
|
Sum: 69,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{3, 0, 9, 12, 21},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{9, 3, 15, 18},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
require.Equal(t, c.expected, c.in.Mul(c.scale))
|
|
// Has it also happened in-place?
|
|
require.Equal(t, c.expected, c.in)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFloatHistogramDiv(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
fh *FloatHistogram
|
|
s float64
|
|
expected *FloatHistogram
|
|
}{
|
|
{
|
|
"zero value",
|
|
&FloatHistogram{},
|
|
3.1415,
|
|
&FloatHistogram{},
|
|
},
|
|
{
|
|
"zero divisor",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: math.Inf(1),
|
|
Count: math.Inf(1),
|
|
Sum: math.Inf(1),
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{math.Inf(1), math.Inf(1), math.Inf(1), math.Inf(1)},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{math.Inf(1), math.Inf(1), math.Inf(1), math.Inf(1)},
|
|
},
|
|
},
|
|
{
|
|
"no-op",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
1,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
},
|
|
{
|
|
"half",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 23,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
2,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 15,
|
|
Sum: 11.5,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{0.5, 0, 1.5, 2, 3.5},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{1.5, 0.5, 2.5, 3},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
require.Equal(t, c.expected, c.fh.Div(c.s))
|
|
// Has it also happened in-place?
|
|
require.Equal(t, c.expected, c.fh)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFloatHistogramDetectReset(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
previous, current *FloatHistogram
|
|
resetExpected bool
|
|
}{
|
|
{
|
|
"zero values",
|
|
&FloatHistogram{},
|
|
&FloatHistogram{},
|
|
false,
|
|
},
|
|
{
|
|
"no buckets to some buckets",
|
|
&FloatHistogram{},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"some buckets to no buckets",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{},
|
|
true,
|
|
},
|
|
{
|
|
"one bucket appears, nothing else changes",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"one bucket disappears, nothing else changes",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"an unpopulated bucket disappears, nothing else changes",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"an unpopulated bucket at the end disappears, nothing else changes",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 1}, {1, 2}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"an unpopulated bucket disappears in a histogram with nothing else",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{23, 1}},
|
|
PositiveBuckets: []float64{0},
|
|
},
|
|
&FloatHistogram{},
|
|
false,
|
|
},
|
|
{
|
|
"zero count goes up",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.6,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"zero count goes down",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.4,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"count goes up",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.4,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"count goes down",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.2,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"sum goes up",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349210,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"sum goes down",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349200,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"one positive bucket goes up",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.3, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"one positive bucket goes down",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.1, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"one negative bucket goes up",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3.1, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"one negative bucket goes down",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 2.9, 1.234e5, 1000},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"zero threshold decreases",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.009,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"zero threshold increases without touching any existing buckets",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.011,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"zero threshold increases enough to cover existing buckets",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 1,
|
|
ZeroCount: 7.73,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{1, 3}},
|
|
PositiveBuckets: []float64{3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"zero threshold increases into the middle of an existing buckets",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.3,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"schema increases without any other changes",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-5, 4}, {2, 6}},
|
|
PositiveBuckets: []float64{0.4, 0.6, 1, 0.23, 2, 1.3, 1.2, 3, 0.05, 0.05},
|
|
NegativeSpans: []Span{{5, 4}, {6, 4}},
|
|
NegativeBuckets: []float64{2, 1.1, 2, 1, 0.234e5, 1e5, 500, 500},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"schema decreases without any other changes",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-5, 4}, {2, 6}},
|
|
PositiveBuckets: []float64{0.4, 0.6, 1, 0.23, 2, 1.3, 1.2, 3, 0.05, 0.05},
|
|
NegativeSpans: []Span{{5, 4}, {6, 4}},
|
|
NegativeBuckets: []float64{2, 1.1, 2, 1, 0.234e5, 1e5, 500, 500},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"schema decreases and a bucket goes up",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-5, 4}, {2, 6}},
|
|
PositiveBuckets: []float64{0.4, 0.6, 1, 0.23, 2, 1.3, 1.2, 3, 0.05, 0.05},
|
|
NegativeSpans: []Span{{5, 4}, {6, 4}},
|
|
NegativeBuckets: []float64{2, 1.1, 2, 1, 0.234e5, 1e5, 500, 500},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 4.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"schema decreases and a bucket goes down",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-5, 4}, {2, 6}},
|
|
PositiveBuckets: []float64{0.4, 0.6, 1, 0.23, 2, 1.3, 1.2, 3, 0.05, 0.05},
|
|
NegativeSpans: []Span{{5, 4}, {6, 4}},
|
|
NegativeBuckets: []float64{2, 1.1, 2, 1, 0.234e5, 1e5, 500, 500},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 5.5,
|
|
Count: 3493.3,
|
|
Sum: 2349209.324,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 1.23, 2.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
true,
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
require.Equal(t, c.resetExpected, c.current.DetectReset(c.previous))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFloatHistogramCompact(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
in *FloatHistogram
|
|
maxEmptyBuckets int
|
|
expected *FloatHistogram
|
|
}{
|
|
{
|
|
"empty histogram",
|
|
&FloatHistogram{},
|
|
0,
|
|
&FloatHistogram{},
|
|
},
|
|
{
|
|
"nothing should happen",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 1}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000},
|
|
},
|
|
},
|
|
{
|
|
"eliminate zero offsets",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 1}, {0, 3}, {0, 1}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {0, 2}, {2, 1}, {0, 1}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 5}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 4}, {2, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"eliminate zero length",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 2}, {2, 0}, {3, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {0, 0}, {2, 0}, {1, 4}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 2}, {5, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 4}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"eliminate multiple zero length spans",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 2}, {2, 0}, {2, 0}, {2, 0}, {3, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 2}, {9, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
},
|
|
},
|
|
{
|
|
"cut empty buckets at start or end",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 4}, {5, 3}},
|
|
PositiveBuckets: []float64{0, 0, 1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 5}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4, 0},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 2}, {5, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 4}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"cut empty buckets at start and end",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 4}, {5, 6}},
|
|
PositiveBuckets: []float64{0, 0, 1, 3.3, 4.2, 0.1, 3.3, 0, 0, 0},
|
|
NegativeSpans: []Span{{-2, 4}, {3, 5}},
|
|
NegativeBuckets: []float64{0, 0, 3.1, 3, 1.234e5, 1000, 3, 4, 0},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 2}, {5, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 4}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"cut empty buckets at start or end of spans, even in the middle",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 6}, {3, 6}},
|
|
PositiveBuckets: []float64{0, 0, 1, 3.3, 0, 0, 4.2, 0.1, 3.3, 0, 0, 0},
|
|
NegativeSpans: []Span{{0, 2}, {2, 6}},
|
|
NegativeBuckets: []float64{3.1, 3, 0, 1.234e5, 1000, 3, 4, 0},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 2}, {5, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 4}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"cut empty buckets at start or end but merge spans due to maxEmptyBuckets",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 4}, {5, 3}},
|
|
PositiveBuckets: []float64{0, 0, 1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 5}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4, 0},
|
|
},
|
|
10,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 10}},
|
|
PositiveBuckets: []float64{1, 3.3, 0, 0, 0, 0, 0, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 9}},
|
|
NegativeBuckets: []float64{3.1, 3, 0, 0, 0, 1.234e5, 1000, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"cut empty buckets from the middle of a span",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 6}, {3, 3}},
|
|
PositiveBuckets: []float64{0, 0, 1, 0, 0, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 5}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 0, 3, 4},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 1}, {2, 1}, {3, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"cut out a span containing only empty buckets",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 3}, {2, 2}, {3, 4}},
|
|
PositiveBuckets: []float64{0, 0, 1, 0, 0, 3.3, 4.2, 0.1, 3.3},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 1}, {7, 4}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
},
|
|
},
|
|
{
|
|
"cut empty buckets from the middle of a span, avoiding some due to maxEmptyBuckets",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 6}, {3, 3}},
|
|
PositiveBuckets: []float64{0, 0, 1, 0, 0, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 5}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 0, 3, 4},
|
|
},
|
|
1,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 1}, {2, 1}, {3, 3}},
|
|
PositiveBuckets: []float64{1, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 5}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 0, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"avoiding all cutting of empty buckets from the middle of a chunk due to maxEmptyBuckets",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 6}, {3, 3}},
|
|
PositiveBuckets: []float64{0, 0, 1, 0, 0, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 5}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 0, 3, 4},
|
|
},
|
|
2,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 4}, {3, 3}},
|
|
PositiveBuckets: []float64{1, 0, 0, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 5}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 0, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"everything merged into one span due to maxEmptyBuckets",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 6}, {3, 3}},
|
|
PositiveBuckets: []float64{0, 0, 1, 0, 0, 3.3, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 2}, {3, 5}},
|
|
NegativeBuckets: []float64{3.1, 3, 1.234e5, 1000, 0, 3, 4},
|
|
},
|
|
3,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-2, 10}},
|
|
PositiveBuckets: []float64{1, 0, 0, 3.3, 0, 0, 0, 4.2, 0.1, 3.3},
|
|
NegativeSpans: []Span{{0, 10}},
|
|
NegativeBuckets: []float64{3.1, 3, 0, 0, 0, 1.234e5, 1000, 0, 3, 4},
|
|
},
|
|
},
|
|
{
|
|
"only empty buckets and maxEmptyBuckets greater zero",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-4, 6}, {3, 3}},
|
|
PositiveBuckets: []float64{0, 0, 0, 0, 0, 0, 0, 0, 0},
|
|
NegativeSpans: []Span{{0, 7}},
|
|
NegativeBuckets: []float64{0, 0, 0, 0, 0, 0, 0},
|
|
},
|
|
3,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{},
|
|
PositiveBuckets: []float64{},
|
|
NegativeSpans: []Span{},
|
|
NegativeBuckets: []float64{},
|
|
},
|
|
},
|
|
{
|
|
"multiple spans of only empty buckets",
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-10, 2}, {2, 1}, {3, 3}},
|
|
PositiveBuckets: []float64{0, 0, 0, 0, 2, 3},
|
|
NegativeSpans: []Span{{-10, 2}, {2, 1}, {3, 3}},
|
|
NegativeBuckets: []float64{2, 3, 0, 0, 0, 0},
|
|
},
|
|
0,
|
|
&FloatHistogram{
|
|
PositiveSpans: []Span{{-1, 2}},
|
|
PositiveBuckets: []float64{2, 3},
|
|
NegativeSpans: []Span{{-10, 2}},
|
|
NegativeBuckets: []float64{2, 3},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
require.Equal(t, c.expected, c.in.Compact(c.maxEmptyBuckets))
|
|
// Compact has happened in-place, too.
|
|
require.Equal(t, c.expected, c.in)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFloatHistogramAdd(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
in1, in2, expected *FloatHistogram
|
|
}{
|
|
{
|
|
"same bucket layout",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{0, 0, 2, 3, 6},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 19,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 5, 7, 13},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{4, 2, 9, 10},
|
|
},
|
|
},
|
|
{
|
|
"same bucket layout, defined differently",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 1}, {0, 2}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 2}, {0, 1}},
|
|
PositiveBuckets: []float64{0, 0, 2, 3, 6},
|
|
NegativeSpans: []Span{{3, 7}},
|
|
NegativeBuckets: []float64{1, 1, 0, 0, 0, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 19,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 1}, {0, 2}},
|
|
PositiveBuckets: []float64{1, 0, 5, 7, 13},
|
|
NegativeSpans: []Span{{3, 5}, {0, 2}},
|
|
NegativeBuckets: []float64{4, 2, 0, 0, 0, 9, 10},
|
|
},
|
|
},
|
|
{
|
|
"non-overlapping spans",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.001,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.001,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{0, 2}, {3, 3}},
|
|
PositiveBuckets: []float64{5, 4, 2, 3, 6},
|
|
NegativeSpans: []Span{{-9, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.001,
|
|
ZeroCount: 19,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-2, 4}, {0, 6}},
|
|
PositiveBuckets: []float64{1, 0, 5, 4, 3, 4, 7, 2, 3, 6},
|
|
NegativeSpans: []Span{{-9, 2}, {3, 2}, {5, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4, 3, 1, 5, 6},
|
|
},
|
|
},
|
|
{
|
|
"non-overlapping inverted order",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{0, 2}, {3, 3}},
|
|
PositiveBuckets: []float64{5, 4, 2, 3, 6},
|
|
NegativeSpans: []Span{{-9, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 19,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-2, 2}, {0, 5}, {0, 3}},
|
|
PositiveBuckets: []float64{1, 0, 5, 4, 3, 4, 7, 2, 3, 6},
|
|
NegativeSpans: []Span{{-9, 2}, {3, 2}, {5, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4, 3, 1, 5, 6},
|
|
},
|
|
},
|
|
{
|
|
"overlapping spans",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{-1, 4}, {0, 3}},
|
|
PositiveBuckets: []float64{5, 4, 2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 19,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-2, 4}, {0, 4}},
|
|
PositiveBuckets: []float64{1, 5, 4, 2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
{
|
|
"overlapping spans inverted order",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{-1, 4}, {0, 3}},
|
|
PositiveBuckets: []float64{5, 4, 2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 19,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-2, 5}, {0, 3}},
|
|
PositiveBuckets: []float64{1, 5, 4, 2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
{
|
|
"schema change",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-1, 4}, {0, 3}},
|
|
PositiveBuckets: []float64{5, 4, 2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-4, 3}, {5, 5}},
|
|
PositiveBuckets: []float64{1, 0, 0, 3, 2, 2, 3, 4},
|
|
NegativeSpans: []Span{{6, 3}, {6, 4}},
|
|
NegativeBuckets: []float64{3, 0.5, 0.5, 2, 3, 2, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 19,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-2, 5}, {0, 3}},
|
|
PositiveBuckets: []float64{1, 5, 4, 2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
{
|
|
"larger zero bucket in first histogram",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 1,
|
|
ZeroCount: 17,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{1, 2}, {0, 3}},
|
|
PositiveBuckets: []float64{2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 1,
|
|
ZeroCount: 29,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{1, 2}, {0, 3}},
|
|
PositiveBuckets: []float64{2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
{
|
|
"larger zero bucket in second histogram",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 1,
|
|
ZeroCount: 17,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{1, 2}, {0, 3}},
|
|
PositiveBuckets: []float64{2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 1,
|
|
ZeroCount: 29,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{1, 5}},
|
|
PositiveBuckets: []float64{2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
{
|
|
"larger zero threshold in first histogram ends up inside a populated bucket of second histogram",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.2,
|
|
ZeroCount: 17,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{1, 2}, {0, 3}},
|
|
PositiveBuckets: []float64{2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.25,
|
|
ZeroCount: 29,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-1, 1}, {1, 5}},
|
|
PositiveBuckets: []float64{0, 2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
{
|
|
"larger zero threshold in second histogram ends up inside a populated bucket of first histogram",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.2,
|
|
ZeroCount: 17,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
PositiveSpans: []Span{{1, 2}, {0, 3}},
|
|
PositiveBuckets: []float64{2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.25,
|
|
ZeroCount: 29,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{1, 5}},
|
|
PositiveBuckets: []float64{2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
{
|
|
"schema change combined with larger zero bucket in second histogram",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-2, 5}, {0, 3}},
|
|
PositiveBuckets: []float64{2, 5, 4, 2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.25,
|
|
ZeroCount: 12,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-3, 2}, {5, 5}},
|
|
PositiveBuckets: []float64{1, 0, 3, 2, 2, 3, 4},
|
|
NegativeSpans: []Span{{6, 3}, {6, 4}},
|
|
NegativeBuckets: []float64{3, 0.5, 0.5, 2, 3, 2, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.25,
|
|
ZeroCount: 22,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-1, 7}},
|
|
PositiveBuckets: []float64{6, 4, 2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
{
|
|
"schema change combined with larger zero bucket in first histogram",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.25,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 1.234,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-1, 4}, {0, 3}},
|
|
PositiveBuckets: []float64{5, 4, 2, 3, 6, 2, 5},
|
|
NegativeSpans: []Span{{4, 2}, {1, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-4, 3}, {5, 5}},
|
|
PositiveBuckets: []float64{1, 0, 0, 3, 2, 2, 3, 4},
|
|
NegativeSpans: []Span{{6, 3}, {6, 4}},
|
|
NegativeBuckets: []float64{3, 0.5, 0.5, 2, 3, 2, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.25,
|
|
ZeroCount: 20,
|
|
Count: 51,
|
|
Sum: 3.579,
|
|
PositiveSpans: []Span{{-1, 4}, {0, 3}},
|
|
PositiveBuckets: []float64{5, 4, 2, 6, 10, 9, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{3, 2, 1, 4, 9, 6},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
require.Equal(t, c.expected, c.in1.Add(c.in2))
|
|
// Has it also happened in-place?
|
|
require.Equal(t, c.expected, c.in1)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFloatHistogramSub(t *testing.T) {
|
|
// This has fewer test cases than TestFloatHistogramAdd because Add and
|
|
// Sub share most of the trickier code.
|
|
cases := []struct {
|
|
name string
|
|
in1, in2, expected *FloatHistogram
|
|
}{
|
|
{
|
|
"same bucket layout",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 23,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 21,
|
|
Sum: 12,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{0, 0, 2, 3, 6},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{1, 1, 4, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 3,
|
|
Count: 9,
|
|
Sum: 11,
|
|
PositiveSpans: []Span{{-2, 2}, {1, 3}},
|
|
PositiveBuckets: []float64{1, 0, 1, 1, 1},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{2, 0, 1, 2},
|
|
},
|
|
},
|
|
{
|
|
"schema change",
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 8,
|
|
Count: 59,
|
|
Sum: 1.234,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-2, 5}, {0, 3}},
|
|
PositiveBuckets: []float64{2, 5, 4, 2, 3, 6, 7, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{4, 10, 1, 4, 14, 7},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 2,
|
|
Count: 19,
|
|
Sum: 0.345,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-4, 3}, {5, 5}},
|
|
PositiveBuckets: []float64{1, 0, 0, 1, 2, 2, 3, 4},
|
|
NegativeSpans: []Span{{6, 3}, {6, 4}},
|
|
NegativeBuckets: []float64{3, 0.5, 0.5, 2, 3, 2, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 6,
|
|
Count: 40,
|
|
Sum: 0.889,
|
|
PositiveSpans: []Span{{-2, 5}, {0, 3}},
|
|
PositiveBuckets: []float64{1, 5, 4, 2, 2, 2, 0, 5},
|
|
NegativeSpans: []Span{{3, 3}, {1, 3}},
|
|
NegativeBuckets: []float64{1, 9, 1, 4, 9, 1},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
require.Equal(t, c.expected, c.in1.Sub(c.in2))
|
|
// Has it also happened in-place?
|
|
require.Equal(t, c.expected, c.in1)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFloatHistogramCopyToSchema(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
targetSchema int32
|
|
in, expected *FloatHistogram
|
|
}{
|
|
{
|
|
"no schema change",
|
|
1,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-4, 3}, {5, 5}},
|
|
PositiveBuckets: []float64{1, 0, 0, 3, 2, 2, 3, 4},
|
|
NegativeSpans: []Span{{6, 3}, {6, 4}},
|
|
NegativeBuckets: []float64{3, 0.5, 0.5, 2, 3, 2, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-4, 3}, {5, 5}},
|
|
PositiveBuckets: []float64{1, 0, 0, 3, 2, 2, 3, 4},
|
|
NegativeSpans: []Span{{6, 3}, {6, 4}},
|
|
NegativeBuckets: []float64{3, 0.5, 0.5, 2, 3, 2, 4},
|
|
},
|
|
},
|
|
{
|
|
"schema change",
|
|
0,
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{{-4, 3}, {5, 5}},
|
|
PositiveBuckets: []float64{1, 0, 0, 3, 2, 2, 3, 4},
|
|
NegativeSpans: []Span{{6, 3}, {6, 4}},
|
|
NegativeBuckets: []float64{3, 0.5, 0.5, 2, 3, 2, 4},
|
|
},
|
|
&FloatHistogram{
|
|
ZeroThreshold: 0.01,
|
|
ZeroCount: 11,
|
|
Count: 30,
|
|
Sum: 2.345,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{{-2, 2}, {2, 3}},
|
|
PositiveBuckets: []float64{1, 0, 3, 4, 7},
|
|
NegativeSpans: []Span{{3, 2}, {3, 2}},
|
|
NegativeBuckets: []float64{3, 1, 5, 6},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
t.Run(c.name, func(t *testing.T) {
|
|
require.Equal(t, c.expected, c.in.CopyToSchema(c.targetSchema))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestReverseFloatBucketIterator(t *testing.T) {
|
|
h := &FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
}
|
|
|
|
// Assuming that the regular iterator is correct.
|
|
|
|
// Positive buckets.
|
|
var expBuckets, actBuckets []Bucket[float64]
|
|
it := h.PositiveBucketIterator()
|
|
for it.Next() {
|
|
// Append in reverse to check reversed list.
|
|
expBuckets = append([]Bucket[float64]{it.At()}, expBuckets...)
|
|
}
|
|
it = h.PositiveReverseBucketIterator()
|
|
for it.Next() {
|
|
actBuckets = append(actBuckets, it.At())
|
|
}
|
|
require.Greater(t, len(expBuckets), 0)
|
|
require.Greater(t, len(actBuckets), 0)
|
|
require.Equal(t, expBuckets, actBuckets)
|
|
|
|
// Negative buckets.
|
|
expBuckets = expBuckets[:0]
|
|
actBuckets = actBuckets[:0]
|
|
it = h.NegativeBucketIterator()
|
|
for it.Next() {
|
|
// Append in reverse to check reversed list.
|
|
expBuckets = append([]Bucket[float64]{it.At()}, expBuckets...)
|
|
}
|
|
it = h.NegativeReverseBucketIterator()
|
|
for it.Next() {
|
|
actBuckets = append(actBuckets, it.At())
|
|
}
|
|
require.Greater(t, len(expBuckets), 0)
|
|
require.Greater(t, len(actBuckets), 0)
|
|
require.Equal(t, expBuckets, actBuckets)
|
|
}
|
|
|
|
func TestAllFloatBucketIterator(t *testing.T) {
|
|
cases := []struct {
|
|
h FloatHistogram
|
|
// To determine the expected buckets.
|
|
includeNeg, includeZero, includePos bool
|
|
}{
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: true,
|
|
includePos: true,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: true,
|
|
includePos: false,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
},
|
|
includeNeg: false,
|
|
includeZero: true,
|
|
includePos: true,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
},
|
|
includeNeg: false,
|
|
includeZero: true,
|
|
includePos: false,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 0,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: false,
|
|
includePos: true,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 447,
|
|
ZeroCount: 42,
|
|
ZeroThreshold: 0.5, // Coinciding with bucket boundary.
|
|
Sum: 1008.4,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: true,
|
|
includePos: true,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 447,
|
|
ZeroCount: 42,
|
|
ZeroThreshold: 0.6, // Within the bucket closest to zero.
|
|
Sum: 1008.4,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: true,
|
|
includePos: true,
|
|
},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
t.Run(fmt.Sprintf("%d", i), func(t *testing.T) {
|
|
var expBuckets, actBuckets []Bucket[float64]
|
|
|
|
if c.includeNeg {
|
|
it := c.h.NegativeReverseBucketIterator()
|
|
for it.Next() {
|
|
b := it.At()
|
|
if c.includeZero && b.Upper > -c.h.ZeroThreshold {
|
|
b.Upper = -c.h.ZeroThreshold
|
|
}
|
|
expBuckets = append(expBuckets, b)
|
|
}
|
|
}
|
|
if c.includeZero {
|
|
expBuckets = append(expBuckets, Bucket[float64]{
|
|
Lower: -c.h.ZeroThreshold,
|
|
Upper: c.h.ZeroThreshold,
|
|
LowerInclusive: true,
|
|
UpperInclusive: true,
|
|
Count: c.h.ZeroCount,
|
|
})
|
|
}
|
|
if c.includePos {
|
|
it := c.h.PositiveBucketIterator()
|
|
for it.Next() {
|
|
b := it.At()
|
|
if c.includeZero && b.Lower < c.h.ZeroThreshold {
|
|
b.Lower = c.h.ZeroThreshold
|
|
}
|
|
expBuckets = append(expBuckets, b)
|
|
}
|
|
}
|
|
|
|
it := c.h.AllBucketIterator()
|
|
for it.Next() {
|
|
actBuckets = append(actBuckets, it.At())
|
|
}
|
|
|
|
require.Equal(t, expBuckets, actBuckets)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestAllReverseFloatBucketIterator(t *testing.T) {
|
|
cases := []struct {
|
|
h FloatHistogram
|
|
// To determine the expected buckets.
|
|
includeNeg, includeZero, includePos bool
|
|
}{
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: true,
|
|
includePos: true,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: true,
|
|
includePos: false,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
},
|
|
includeNeg: false,
|
|
includeZero: true,
|
|
includePos: true,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 102,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
},
|
|
includeNeg: false,
|
|
includeZero: true,
|
|
includePos: false,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 405,
|
|
ZeroCount: 0,
|
|
ZeroThreshold: 0.001,
|
|
Sum: 1008.4,
|
|
Schema: 1,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: false,
|
|
includePos: true,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 447,
|
|
ZeroCount: 42,
|
|
ZeroThreshold: 0.5, // Coinciding with bucket boundary.
|
|
Sum: 1008.4,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: true,
|
|
includePos: true,
|
|
},
|
|
{
|
|
h: FloatHistogram{
|
|
Count: 447,
|
|
ZeroCount: 42,
|
|
ZeroThreshold: 0.6, // Within the bucket closest to zero.
|
|
Sum: 1008.4,
|
|
Schema: 0,
|
|
PositiveSpans: []Span{
|
|
{Offset: 0, Length: 4},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 3},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
PositiveBuckets: []float64{100, 344, 123, 55, 3, 63, 2, 54, 235, 33},
|
|
NegativeSpans: []Span{
|
|
{Offset: 0, Length: 3},
|
|
{Offset: 1, Length: 0},
|
|
{Offset: 3, Length: 0},
|
|
{Offset: 3, Length: 4},
|
|
{Offset: 2, Length: 0},
|
|
{Offset: 5, Length: 3},
|
|
},
|
|
NegativeBuckets: []float64{10, 34, 1230, 54, 67, 63, 2, 554, 235, 33},
|
|
},
|
|
includeNeg: true,
|
|
includeZero: true,
|
|
includePos: true,
|
|
},
|
|
}
|
|
|
|
for i, c := range cases {
|
|
t.Run(fmt.Sprintf("%d", i), func(t *testing.T) {
|
|
var expBuckets, actBuckets []Bucket[float64]
|
|
|
|
if c.includePos {
|
|
it := c.h.PositiveReverseBucketIterator()
|
|
for it.Next() {
|
|
b := it.At()
|
|
if c.includeZero && b.Lower < c.h.ZeroThreshold {
|
|
b.Lower = c.h.ZeroThreshold
|
|
}
|
|
expBuckets = append(expBuckets, b)
|
|
}
|
|
}
|
|
if c.includeZero {
|
|
expBuckets = append(expBuckets, Bucket[float64]{
|
|
Lower: -c.h.ZeroThreshold,
|
|
Upper: c.h.ZeroThreshold,
|
|
LowerInclusive: true,
|
|
UpperInclusive: true,
|
|
Count: c.h.ZeroCount,
|
|
})
|
|
}
|
|
if c.includeNeg {
|
|
it := c.h.NegativeBucketIterator()
|
|
for it.Next() {
|
|
b := it.At()
|
|
if c.includeZero && b.Upper > -c.h.ZeroThreshold {
|
|
b.Upper = -c.h.ZeroThreshold
|
|
}
|
|
expBuckets = append(expBuckets, b)
|
|
}
|
|
}
|
|
|
|
it := c.h.AllReverseBucketIterator()
|
|
for it.Next() {
|
|
actBuckets = append(actBuckets, it.At())
|
|
}
|
|
|
|
require.Equal(t, expBuckets, actBuckets)
|
|
})
|
|
}
|
|
}
|