mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-20 23:46:23 +01:00
650 lines
20 KiB
Go
650 lines
20 KiB
Go
|
package promql
|
||
|
|
||
|
import (
|
||
|
"math"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
testValues = []float64{123, 34, 44, 21, 54, 34, 99, 12, 44, 32, 34, 34}
|
||
|
testTimestamps = []int64{5, 15, 24, 36, 49, 60, 78, 80, 97, 115, 120, 130}
|
||
|
)
|
||
|
|
||
|
func TestRemoveCounterResets(t *testing.T) {
|
||
|
removeCounterResets(nil)
|
||
|
|
||
|
values := append([]float64{}, testValues...)
|
||
|
removeCounterResets(values)
|
||
|
valuesExpected := []float64{123, 157, 167, 188, 221, 255, 320, 332, 364, 396, 398, 398}
|
||
|
testRowsEqual(t, values, testTimestamps, valuesExpected, testTimestamps)
|
||
|
|
||
|
// removeCounterResets doesn't expect negative values, so it doesn't work properly with them.
|
||
|
values = []float64{-100, -200, -300, -400}
|
||
|
removeCounterResets(values)
|
||
|
valuesExpected = []float64{-100, -300, -600, -1000}
|
||
|
timestampsExpected := []int64{0, 1, 2, 3}
|
||
|
testRowsEqual(t, values, timestampsExpected, valuesExpected, timestampsExpected)
|
||
|
|
||
|
// verify how jitter from `Prometheus HA pairs` is handled
|
||
|
values = []float64{100, 95, 120, 140, 137, 50}
|
||
|
removeCounterResets(values)
|
||
|
valuesExpected = []float64{100, 100, 120, 140, 140, 190}
|
||
|
timestampsExpected = []int64{0, 1, 2, 3, 4, 5}
|
||
|
testRowsEqual(t, values, timestampsExpected, valuesExpected, timestampsExpected)
|
||
|
}
|
||
|
|
||
|
func TestDeltaValues(t *testing.T) {
|
||
|
deltaValues(nil)
|
||
|
|
||
|
values := []float64{123}
|
||
|
deltaValues(values)
|
||
|
valuesExpected := []float64{nan}
|
||
|
testRowsEqual(t, values, testTimestamps[:1], valuesExpected, testTimestamps[:1])
|
||
|
|
||
|
values = append([]float64{}, testValues...)
|
||
|
deltaValues(values)
|
||
|
valuesExpected = []float64{-89, 10, -23, 33, -20, 65, -87, 32, -12, 2, 0, nan}
|
||
|
testRowsEqual(t, values, testTimestamps, valuesExpected, testTimestamps)
|
||
|
|
||
|
// remove counter resets
|
||
|
values = append([]float64{}, testValues...)
|
||
|
removeCounterResets(values)
|
||
|
deltaValues(values)
|
||
|
valuesExpected = []float64{34, 10, 21, 33, 34, 65, 12, 32, 32, 2, 0, nan}
|
||
|
testRowsEqual(t, values, testTimestamps, valuesExpected, testTimestamps)
|
||
|
}
|
||
|
|
||
|
func TestDerivValues(t *testing.T) {
|
||
|
derivValues(nil, nil)
|
||
|
|
||
|
values := []float64{123}
|
||
|
derivValues(values, testTimestamps[:1])
|
||
|
valuesExpected := []float64{nan}
|
||
|
testRowsEqual(t, values, testTimestamps[:1], valuesExpected, testTimestamps[:1])
|
||
|
|
||
|
values = append([]float64{}, testValues...)
|
||
|
derivValues(values, testTimestamps)
|
||
|
valuesExpected = []float64{-8900, 1111.111111111111, -1916.6666666666665, 2538.461538461538, -1818.1818181818182, 3611.111111111111,
|
||
|
-43500, 1882.3529411764705, -666.6666666666666, 400, 0, nan}
|
||
|
testRowsEqual(t, values, testTimestamps, valuesExpected, testTimestamps)
|
||
|
|
||
|
// remove counter resets
|
||
|
values = append([]float64{}, testValues...)
|
||
|
removeCounterResets(values)
|
||
|
derivValues(values, testTimestamps)
|
||
|
valuesExpected = []float64{3400, 1111.111111111111, 1750, 2538.461538461538, 3090.909090909091, 3611.111111111111,
|
||
|
6000, 1882.3529411764705, 1777.7777777777776, 400, 0, nan}
|
||
|
testRowsEqual(t, values, testTimestamps, valuesExpected, testTimestamps)
|
||
|
}
|
||
|
|
||
|
func testRollupFunc(t *testing.T, funcName string, args []interface{}, meExpected *metricExpr, vExpected float64) {
|
||
|
t.Helper()
|
||
|
nrf := getRollupFunc(funcName)
|
||
|
if nrf == nil {
|
||
|
t.Fatalf("cannot obtain %q", funcName)
|
||
|
}
|
||
|
rf, err := nrf(args)
|
||
|
if err != nil {
|
||
|
t.Fatalf("unexpected error: %s", err)
|
||
|
}
|
||
|
var rfa rollupFuncArg
|
||
|
rfa.prevValue = nan
|
||
|
rfa.prevTimestamp = 0
|
||
|
rfa.values = append(rfa.values, testValues...)
|
||
|
rfa.timestamps = append(rfa.timestamps, testTimestamps...)
|
||
|
if rollupFuncsRemoveCounterResets[funcName] {
|
||
|
removeCounterResets(rfa.values)
|
||
|
}
|
||
|
for i := 0; i < 5; i++ {
|
||
|
v := rf(&rfa)
|
||
|
if math.IsNaN(vExpected) {
|
||
|
if !math.IsNaN(v) {
|
||
|
t.Fatalf("unexpected value; got %v; want %v", v, vExpected)
|
||
|
}
|
||
|
} else {
|
||
|
if v != vExpected {
|
||
|
t.Fatalf("unexpected value; got %v; want %v", v, vExpected)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestRollupQuantileOverTime(t *testing.T) {
|
||
|
f := func(phi, vExpected float64) {
|
||
|
t.Helper()
|
||
|
phis := []*timeseries{{
|
||
|
Values: []float64{phi},
|
||
|
Timestamps: []int64{123},
|
||
|
}}
|
||
|
var me metricExpr
|
||
|
args := []interface{}{phis, &rollupExpr{Expr: &me}}
|
||
|
testRollupFunc(t, "quantile_over_time", args, &me, vExpected)
|
||
|
}
|
||
|
|
||
|
f(-123, 12)
|
||
|
f(-0.5, 12)
|
||
|
f(0, 12)
|
||
|
f(0.1, 21)
|
||
|
f(0.5, 34)
|
||
|
f(0.9, 99)
|
||
|
f(1, 123)
|
||
|
f(234, 123)
|
||
|
}
|
||
|
|
||
|
func TestRollupPredictLinear(t *testing.T) {
|
||
|
f := func(sec, vExpected float64) {
|
||
|
t.Helper()
|
||
|
secs := []*timeseries{{
|
||
|
Values: []float64{sec},
|
||
|
Timestamps: []int64{123},
|
||
|
}}
|
||
|
var me metricExpr
|
||
|
args := []interface{}{&rollupExpr{Expr: &me}, secs}
|
||
|
testRollupFunc(t, "predict_linear", args, &me, vExpected)
|
||
|
}
|
||
|
|
||
|
f(0e-3, 63.739757761102624)
|
||
|
f(50e-3, 50.39682764539959)
|
||
|
f(100e-3, 37.053897529696556)
|
||
|
f(200e-3, 10.368037298290488)
|
||
|
}
|
||
|
|
||
|
func TestRollupHoltWinters(t *testing.T) {
|
||
|
f := func(sf, tf, vExpected float64) {
|
||
|
t.Helper()
|
||
|
sfs := []*timeseries{{
|
||
|
Values: []float64{sf},
|
||
|
Timestamps: []int64{123},
|
||
|
}}
|
||
|
tfs := []*timeseries{{
|
||
|
Values: []float64{tf},
|
||
|
Timestamps: []int64{123},
|
||
|
}}
|
||
|
var me metricExpr
|
||
|
args := []interface{}{&rollupExpr{Expr: &me}, sfs, tfs}
|
||
|
testRollupFunc(t, "holt_winters", args, &me, vExpected)
|
||
|
}
|
||
|
|
||
|
f(-1, 0.5, nan)
|
||
|
f(0, 0.5, nan)
|
||
|
f(1, 0.5, nan)
|
||
|
f(2, 0.5, nan)
|
||
|
f(0.5, -1, nan)
|
||
|
f(0.5, 0, nan)
|
||
|
f(0.5, 1, nan)
|
||
|
f(0.5, 2, nan)
|
||
|
f(0.5, 0.5, 34.97794532775879)
|
||
|
f(0.1, 0.5, -131.30529492371622)
|
||
|
f(0.1, 0.1, -397.3307790780296)
|
||
|
f(0.5, 0.1, -5.791530520284198)
|
||
|
f(0.5, 0.9, 25.498906408926757)
|
||
|
f(0.9, 0.9, 33.99637566941818)
|
||
|
}
|
||
|
|
||
|
func TestRollupNewRollupFuncSuccess(t *testing.T) {
|
||
|
f := func(funcName string, vExpected float64) {
|
||
|
t.Helper()
|
||
|
var me metricExpr
|
||
|
args := []interface{}{&rollupExpr{Expr: &me}}
|
||
|
testRollupFunc(t, funcName, args, &me, vExpected)
|
||
|
}
|
||
|
|
||
|
f("default_rollup", 123)
|
||
|
f("changes", 10)
|
||
|
f("delta", -89)
|
||
|
f("deriv", -712)
|
||
|
f("idelta", 0)
|
||
|
f("increase", 275)
|
||
|
f("irate", 0)
|
||
|
f("rate", 2200)
|
||
|
f("resets", 5)
|
||
|
f("avg_over_time", 47.083333333333336)
|
||
|
f("min_over_time", 12)
|
||
|
f("max_over_time", 123)
|
||
|
f("sum_over_time", 565)
|
||
|
f("count_over_time", 12)
|
||
|
f("stddev_over_time", 30.752935722554287)
|
||
|
f("stdvar_over_time", 945.7430555555555)
|
||
|
f("first_over_time", 123)
|
||
|
f("last_over_time", 34)
|
||
|
f("integrate", 61.0275)
|
||
|
}
|
||
|
|
||
|
func TestRollupNewRollupFuncError(t *testing.T) {
|
||
|
if nrf := getRollupFunc("non-existing-func"); nrf != nil {
|
||
|
t.Fatalf("expecting nil func; got %p", nrf)
|
||
|
}
|
||
|
|
||
|
f := func(funcName string, args []interface{}) {
|
||
|
t.Helper()
|
||
|
|
||
|
nrf := getRollupFunc(funcName)
|
||
|
rf, err := nrf(args)
|
||
|
if err == nil {
|
||
|
t.Fatalf("expecting non-nil error")
|
||
|
}
|
||
|
if rf != nil {
|
||
|
t.Fatalf("expecting nil rf; got %p", rf)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Invalid number of args
|
||
|
f("default_rollup", nil)
|
||
|
f("holt_winters", nil)
|
||
|
f("predict_linear", nil)
|
||
|
f("quantile_over_time", nil)
|
||
|
|
||
|
// Invalid arg type
|
||
|
scalarTs := []*timeseries{{
|
||
|
Values: []float64{321},
|
||
|
Timestamps: []int64{123},
|
||
|
}}
|
||
|
me := &metricExpr{}
|
||
|
f("holt_winters", []interface{}{123, 123, 321})
|
||
|
f("holt_winters", []interface{}{me, 123, 321})
|
||
|
f("holt_winters", []interface{}{me, scalarTs, 321})
|
||
|
f("predict_linear", []interface{}{123, 123})
|
||
|
f("predict_linear", []interface{}{me, 123})
|
||
|
f("quantile_over_time", []interface{}{123, 123})
|
||
|
}
|
||
|
|
||
|
func TestRollupNoWindowNoPoints(t *testing.T) {
|
||
|
t.Run("beforeStart", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupFirst,
|
||
|
Start: 0,
|
||
|
End: 4,
|
||
|
Step: 1,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{nan, nan, nan, nan, 123}
|
||
|
timestampsExpected := []int64{0, 1, 2, 3, 4}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("afterEnd", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupDelta,
|
||
|
Start: 120,
|
||
|
End: 144,
|
||
|
Step: 4,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{2, 2, 0, 0, 0, nan, nan}
|
||
|
timestampsExpected := []int64{120, 124, 128, 132, 136, 140, 144}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestRollupWindowNoPoints(t *testing.T) {
|
||
|
t.Run("beforeStart", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupFirst,
|
||
|
Start: 0,
|
||
|
End: 4,
|
||
|
Step: 1,
|
||
|
Window: 3,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{nan, nan, nan, nan, 123}
|
||
|
timestampsExpected := []int64{0, 1, 2, 3, 4}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("afterEnd", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupFirst,
|
||
|
Start: 141,
|
||
|
End: 171,
|
||
|
Step: 10,
|
||
|
Window: 3,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{34, nan, nan, nan}
|
||
|
timestampsExpected := []int64{141, 151, 161, 171}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestRollupNoWindowPartialPoints(t *testing.T) {
|
||
|
t.Run("beforeStart", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupFirst,
|
||
|
Start: 0,
|
||
|
End: 20,
|
||
|
Step: 5,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{123, 123, 123, 123, 123}
|
||
|
timestampsExpected := []int64{0, 5, 10, 15, 20}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("afterEnd", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupFirst,
|
||
|
Start: 100,
|
||
|
End: 160,
|
||
|
Step: 20,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{44, 34, 34, nan}
|
||
|
timestampsExpected := []int64{100, 120, 140, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("middle", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupFirst,
|
||
|
Start: -50,
|
||
|
End: 150,
|
||
|
Step: 50,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{nan, 123, 54, 44, nan}
|
||
|
timestampsExpected := []int64{-50, 0, 50, 100, 150}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestRollupWindowPartialPoints(t *testing.T) {
|
||
|
t.Run("beforeStart", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupLast,
|
||
|
Start: 0,
|
||
|
End: 20,
|
||
|
Step: 5,
|
||
|
Window: 8,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{123, 123, 34, 34, 44}
|
||
|
timestampsExpected := []int64{0, 5, 10, 15, 20}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("afterEnd", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupLast,
|
||
|
Start: 100,
|
||
|
End: 160,
|
||
|
Step: 20,
|
||
|
Window: 18,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{34, 34, nan, nan}
|
||
|
timestampsExpected := []int64{100, 120, 140, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("middle", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupLast,
|
||
|
Start: 0,
|
||
|
End: 150,
|
||
|
Step: 50,
|
||
|
Window: 19,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{54, 44, nan, nan}
|
||
|
timestampsExpected := []int64{0, 50, 100, 150}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestRollupFuncsNoWindow(t *testing.T) {
|
||
|
t.Run("first", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupFirst,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{123, 21, 12, 34, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("count", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupCount,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{4, 4, 3, 1, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("min", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupMin,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{21, 12, 32, 34, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("max", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupMax,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{123, 99, 44, 34, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("sum", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupSum,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{222, 199, 110, 34, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("delta", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupDelta,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{-102, -9, 22, 0, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("idelta", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupIdelta,
|
||
|
Start: 10,
|
||
|
End: 130,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{33, -87, 0, nan}
|
||
|
timestampsExpected := []int64{10, 50, 90, 130}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("changes", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupChanges,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{3, 4, 3, 0, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("resets", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupResets,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{2, 2, 1, 0, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("avg", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupAvg,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{55.5, 49.75, 36.666666666666664, 34, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("deriv", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupDeriv,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{-3290.3225806451615, -204.54545454545456, 550, 0, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("ideriv", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupIderiv,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{-1916.6666666666665, -43500, 400, 0, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("stddev", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupStddev,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{39.81519810323691, 32.080952292598795, 5.2493385826745405, 0, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("integrate", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupIntegrate,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{4.6035, 4.3934999999999995, 2.166, 0.34, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
t.Run("distinct", func(t *testing.T) {
|
||
|
rc := rollupConfig{
|
||
|
Func: rollupDistinct,
|
||
|
Start: 0,
|
||
|
End: 160,
|
||
|
Step: 40,
|
||
|
Window: 0,
|
||
|
}
|
||
|
rc.Timestamps = getTimestamps(rc.Start, rc.End, rc.Step)
|
||
|
values := rc.Do(nil, testValues, testTimestamps)
|
||
|
valuesExpected := []float64{4, 4, 3, 1, nan}
|
||
|
timestampsExpected := []int64{0, 40, 80, 120, 160}
|
||
|
testRowsEqual(t, values, rc.Timestamps, valuesExpected, timestampsExpected)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func testRowsEqual(t *testing.T, values []float64, timestamps []int64, valuesExpected []float64, timestampsExpected []int64) {
|
||
|
t.Helper()
|
||
|
if len(values) != len(valuesExpected) {
|
||
|
t.Fatalf("unexpected len(values); got %d; want %d\nvalues=\n%v\nvaluesExpected=\n%v",
|
||
|
len(values), len(valuesExpected), values, valuesExpected)
|
||
|
}
|
||
|
if len(timestamps) != len(timestampsExpected) {
|
||
|
t.Fatalf("unexpected len(timestamps); got %d; want %d\ntimestamps=\n%v\ntimestampsExpected=\n%v",
|
||
|
len(timestamps), len(timestampsExpected), timestamps, timestampsExpected)
|
||
|
}
|
||
|
if len(values) != len(timestamps) {
|
||
|
t.Fatalf("len(values) doesn't match len(timestamps); got %d vs %d", len(values), len(timestamps))
|
||
|
}
|
||
|
for i, v := range values {
|
||
|
ts := timestamps[i]
|
||
|
tsExpected := timestampsExpected[i]
|
||
|
if ts != tsExpected {
|
||
|
t.Fatalf("unexpected timestamp at timestamps[%d]; got %d; want %d\ntimestamps=\n%v\ntimestampsExpected=\n%v",
|
||
|
i, ts, tsExpected, timestamps, timestampsExpected)
|
||
|
}
|
||
|
vExpected := valuesExpected[i]
|
||
|
if math.IsNaN(v) {
|
||
|
if !math.IsNaN(vExpected) {
|
||
|
t.Fatalf("unexpected nan value at values[%d]; want %f\nvalues=\n%v\nvaluesExpected=\n%v",
|
||
|
i, vExpected, values, valuesExpected)
|
||
|
}
|
||
|
continue
|
||
|
}
|
||
|
if v != vExpected {
|
||
|
t.Fatalf("unexpected value at values[%d]; got %f; want %f\nvalues=\n%v\nvaluesExpected=\n%v",
|
||
|
i, v, vExpected, values, valuesExpected)
|
||
|
}
|
||
|
}
|
||
|
}
|