113 lines
2.5 KiB
Go
113 lines
2.5 KiB
Go
package pmap
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func TestPmap(t *testing.T) {
|
|
p := New()
|
|
const numElems = 100
|
|
for i := range iter(numElems) {
|
|
p = p.Set(i, i)
|
|
}
|
|
if p.Len() != numElems {
|
|
t.Fatalf("Len() = %v, want %v", p.Len(), numElems)
|
|
}
|
|
for i := range iter(numElems) {
|
|
v, ok := p.Get(i)
|
|
if v != i || !ok {
|
|
t.Errorf("Get(%d) = %v %v, want %v %v", i, v, ok, i, true)
|
|
}
|
|
}
|
|
fmt.Print(p.(pmap).stats())
|
|
}
|
|
|
|
func BenchmarkGet(b *testing.B) {
|
|
b.Run("size=10", func(b *testing.B) { benchmarkGet(b, 10) })
|
|
b.Run("size=100", func(b *testing.B) { benchmarkGet(b, 100) })
|
|
b.Run("size=1000", func(b *testing.B) { benchmarkGet(b, 1000) })
|
|
b.Run("size=10000", func(b *testing.B) { benchmarkGet(b, 10000) })
|
|
}
|
|
|
|
func BenchmarkGetAbsent(b *testing.B) {
|
|
b.Run("size=10", func(b *testing.B) { benchmarkGetAbsent(b, 10) })
|
|
b.Run("size=100", func(b *testing.B) { benchmarkGetAbsent(b, 100) })
|
|
b.Run("size=1000", func(b *testing.B) { benchmarkGetAbsent(b, 1000) })
|
|
b.Run("size=10000", func(b *testing.B) { benchmarkGetAbsent(b, 10000) })
|
|
}
|
|
|
|
func benchmarkGet(b *testing.B, numElems int) {
|
|
p := New()
|
|
for i := range iter(numElems) {
|
|
p = p.Set(i, i)
|
|
}
|
|
if p.Len() != numElems {
|
|
b.Fatalf("Len() = %v, want %v", p.Len(), numElems)
|
|
}
|
|
b.ResetTimer()
|
|
i := 0
|
|
for range iter(b.N) {
|
|
v, ok := p.Get(i)
|
|
if v != i || !ok {
|
|
b.Errorf("Get(%d) = %v %v, want %v %v", i, v, ok, i, true)
|
|
}
|
|
i++
|
|
if i >= numElems {
|
|
i = 0
|
|
}
|
|
}
|
|
}
|
|
|
|
func benchmarkGetAbsent(b *testing.B, numElems int) {
|
|
p := New()
|
|
for i := range iter(numElems) {
|
|
p = p.Set(-i-1, i)
|
|
}
|
|
if p.Len() != numElems {
|
|
b.Fatalf("Len() = %v, want %v", p.Len(), numElems)
|
|
}
|
|
b.ResetTimer()
|
|
i := 0
|
|
for range iter(b.N) {
|
|
v, ok := p.Get(i)
|
|
if ok {
|
|
b.Errorf("Get(%d) = %v %v, want %v %v", i, v, ok, 0, false)
|
|
}
|
|
i++
|
|
}
|
|
}
|
|
|
|
func iter(n int) []struct{} {
|
|
return make([]struct{}, n)
|
|
}
|
|
|
|
func BenchmarkHmapGet_baseline(b *testing.B) {
|
|
b.Run("size=10", func(b *testing.B) { benchmarkHmapGet(b, 10) })
|
|
b.Run("size=100", func(b *testing.B) { benchmarkHmapGet(b, 100) })
|
|
b.Run("size=1000", func(b *testing.B) { benchmarkHmapGet(b, 1000) })
|
|
b.Run("size=10000", func(b *testing.B) { benchmarkHmapGet(b, 10000) })
|
|
}
|
|
|
|
func benchmarkHmapGet(b *testing.B, numElems int) {
|
|
h := make(map[int]int)
|
|
for i := range iter(numElems) {
|
|
h[i] = i
|
|
}
|
|
if len(h) != numElems {
|
|
b.Fatalf("Len() = %v, want %v", len(h), numElems)
|
|
}
|
|
b.ResetTimer()
|
|
i := 0
|
|
for range iter(b.N) {
|
|
v, ok := h[i]
|
|
if v != i || !ok {
|
|
b.Errorf("h[%d] = %v %v, want %v %v", i, v, ok, i, true)
|
|
}
|
|
i++
|
|
if i >= numElems {
|
|
i = 0
|
|
}
|
|
}
|
|
}
|