container/sampler/weighted_linear_test.go

// Copyright (C) 2019-2025, Lux Industries, Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package sampler

import (
	"fmt"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestWeightedLinearElementCompare(t *testing.T) {
	tests := []struct {
		a        weightedLinearElement
		b        weightedLinearElement
		expected int
	}{
		{
			a:        weightedLinearElement{},
			b:        weightedLinearElement{},
			expected: 0,
		},
		{
			a: weightedLinearElement{
				cumulativeWeight: 1,
			},
			b: weightedLinearElement{
				cumulativeWeight: 2,
			},
			expected: 1,
		},
	}
	for _, test := range tests {
		t.Run(fmt.Sprintf("%d_%d_%d", test.a.cumulativeWeight, test.b.cumulativeWeight, test.expected), func(t *testing.T) {
			require := require.New(t)

			require.Equal(test.expected, test.a.Compare(test.b))
			require.Equal(-test.expected, test.b.Compare(test.a))
		})
	}
}