mirror of
https://github.com/luxfi/container.git
synced 2026-09-24 00:24:34 +00:00
zeekayandHanzo Dev
adf4508709
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers: any, WaitGroup.Go, range over int, and a promoted-field key in a struct literal. omitzero is not applied, since it changes what the JSON says. Co-authored-by: Hanzo Dev <[email protected]>
67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
// Copyright (C) 2019-2025, Lux Industries, Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package sampler
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
safemath "github.com/luxfi/math"
|
|
)
|
|
|
|
func TestWeightedInitializeOverflow(t *testing.T) {
|
|
s := &weightedHeap{}
|
|
err := s.Initialize([]uint64{1, math.MaxUint64})
|
|
require.ErrorIs(t, err, safemath.ErrOverflow)
|
|
}
|
|
|
|
func TestWeightedOutOfRange(t *testing.T) {
|
|
require := require.New(t)
|
|
s := &weightedHeap{}
|
|
|
|
require.NoError(s.Initialize([]uint64{1}))
|
|
|
|
_, ok := s.Sample(1)
|
|
require.False(ok)
|
|
}
|
|
|
|
func TestWeightedSingleton(t *testing.T) {
|
|
require := require.New(t)
|
|
s := &weightedHeap{}
|
|
|
|
require.NoError(s.Initialize([]uint64{1}))
|
|
|
|
index, ok := s.Sample(0)
|
|
require.True(ok)
|
|
require.Zero(index)
|
|
}
|
|
|
|
func TestWeightedWithZero(t *testing.T) {
|
|
require := require.New(t)
|
|
s := &weightedHeap{}
|
|
|
|
require.NoError(s.Initialize([]uint64{0, 1}))
|
|
|
|
index, ok := s.Sample(0)
|
|
require.True(ok)
|
|
require.Equal(1, index)
|
|
}
|
|
|
|
func TestWeightedDistribution(t *testing.T) {
|
|
require := require.New(t)
|
|
s := &weightedHeap{}
|
|
|
|
require.NoError(s.Initialize([]uint64{1, 1, 2, 3, 4}))
|
|
|
|
counts := make([]int, 5)
|
|
for i := range uint64(11) {
|
|
index, ok := s.Sample(i)
|
|
require.True(ok)
|
|
counts[index]++
|
|
}
|
|
require.Equal([]int{1, 1, 2, 3, 4}, counts)
|
|
}
|