math/set/set_benchmark_test.go

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

package set

import (
	"strconv"
	"testing"
)

func BenchmarkSetList(b *testing.B) {
	sizes := []int{5, 25, 100, 100_000} // Test with various sizes
	for size := range sizes {
		b.Run(strconv.Itoa(size), func(b *testing.B) {
			set := Set[int]{}
			for i := 0; i < size; i++ {
				set.Add(i)
			}
			b.ResetTimer()
			for n := 0; n < b.N; n++ {
				set.List()
			}
		})
	}
}

func BenchmarkSetClear(b *testing.B) {
	for _, numElts := range []int{10, 25, 50, 100, 250, 500, 1000} {
		b.Run(strconv.Itoa(numElts), func(b *testing.B) {
			set := NewSet[int](numElts)
			for n := 0; n < b.N; n++ {
				for i := 0; i < numElts; i++ {
					set.Add(i)
				}
				set.Clear()
			}
		})
	}
}