Function TestAvgOverTimeScalesResults

pkg/traceql/engine_metrics_average_test.go:124–229

(t *testing.T)

}

func TestAvgOverTimeScalesResults(t *testing.T) {
	// Test data designed to verify multiplication accuracy in weighted mean calculation
	// The addWeigthedMean function uses: meanDelta := ((mean - currentMean.mean) * weight) / sumWeights
	in := []Span{
		// Time interval 1: service=a has 2 spans with durations 100, 200 (mean=150, count=2)
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("service", "a").WithDuration(uint64(100 * time.Second)),
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("service", "a").WithDuration(uint64(200 * time.Second)),

		// Time interval 1: service=b has 3 spans with durations 300, 600, 900 (mean=600, count=3)
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("service", "b").WithDuration(uint64(300 * time.Second)),
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("service", "b").WithDuration(uint64(600 * time.Second)),
		newMockSpan(nil).WithStartTime(uint64(1*time.Second)).WithSpanString("service", "b").WithDuration(uint64(900 * time.Second)),

		// Time interval 2: service=a has 4 spans with durations 400, 800, 1200, 1600 (mean=1000, count=4)
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("service", "a").WithDuration(uint64(400 * time.Second)),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("service", "a").WithDuration(uint64(800 * time.Second)),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("service", "a").WithDuration(uint64(1200 * time.Second)),
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("service", "a").WithDuration(uint64(1600 * time.Second)),

		// Time interval 2: service=b has 1 span with duration 2000 (mean=2000, count=1)
		newMockSpan(nil).WithStartTime(uint64(2*time.Second)).WithSpanString("service", "b").WithDuration(uint64(2000 * time.Second)),

		// Time interval 3: service=a has varying precision values to test floating point accuracy
		newMockSpan(nil).WithStartTime(uint64(3*time.Second)).WithSpanString("service", "a").WithDuration(uint64(123 * time.Second)),
		newMockSpan(nil).WithStartTime(uint64(3*time.Second)).WithSpanString("service", "a").WithDuration(uint64(456 * time.Second)),
		newMockSpan(nil).WithStartTime(uint64(3*time.Second)).WithSpanString("service", "a").WithDuration(uint64(789 * time.Second)),
	}

	a := newAverageOverTimeMetricsAggregator(IntrinsicDurationAttribute, []Attribute{NewAttribute("service")})
	a.init(&tempopb.QueryRangeRequest{
		Start: 1,
		End:   uint64(3 * time.Second),
		Step:  uint64(1 * time.Second),
	}, AggregateModeRaw)

	for _, span := range in {
		a.observe(span)
	}

	ss := a.result(1.0)
	expected := []TimeSeries{
		{
			Labels: Labels{
				Label{Name: ".service", Value: NewStaticString("a")},
			},
			Values:    []float64{150.0, 1000.0, 456.0},
			Exemplars: []Exemplar{},
		},
		{
			Labels: Labels{
				Label{Name: ".service", Value: NewStaticString("a")},
				Label{internalLabelMetaType, NewStaticString(internalMetaTypeCount)},
			},
			Values: []float64{2, 4, 3},
		},
		{
			Labels: Labels{
				Label{Name: ".service", Value: NewStaticString("b")},