golang
/
set
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

test: fix benchmark

This commit is contained in:
Nao Yonashiro 2022-03-28 17:39:59 +09:00 committed by Ralph Caraveo
parent 42e82971af
commit abcdb7d406
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
bench_test.go
View File

@ -38,20 +38,23 @@ func nrand(n int) []int {
return i
}
func benchAdd(b *testing.B, s Set[int]) {
nums := nrand(b.N)
func benchAdd(b *testing.B, n int, newSet func(...int) Set[int]) {
nums := nrand(n)
b.ResetTimer()
for _, v := range nums {
s.Add(v)
for i := 0; i < b.N; i++ {
s := newSet()
for _, v := range nums {
s.Add(v)
}
}
}
func BenchmarkAddSafe(b *testing.B) {
benchAdd(b, NewSet[int]())
benchAdd(b, 1000, NewSet[int])
}
func BenchmarkAddUnsafe(b *testing.B) {
benchAdd(b, NewThreadUnsafeSet[int]())
benchAdd(b, 1000, NewThreadUnsafeSet[int])
}
func benchRemove(b *testing.B, s Set[int]) {