53 lines
935 B
Go
53 lines
935 B
Go
package hashmap
|
|
|
|
import (
|
|
"fmt"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"474420502.top/eson/structure/compare"
|
|
)
|
|
|
|
func TestCount(t *testing.T) {
|
|
hm := New(HashInt, compare.Int)
|
|
|
|
for i := 0; i < 1000; i++ {
|
|
hm.Put(i, i)
|
|
}
|
|
}
|
|
|
|
func PrintMemUsage() {
|
|
var m runtime.MemStats
|
|
runtime.ReadMemStats(&m)
|
|
// For info on each, see: https://golang.org/pkg/runtime/#MemStats
|
|
fmt.Printf("Alloc = %v MiB", bToMb(m.Alloc))
|
|
fmt.Printf("\tTotalAlloc = %v MiB", bToMb(m.TotalAlloc))
|
|
fmt.Printf("\tSys = %v MiB", bToMb(m.Sys))
|
|
fmt.Printf("\tNumGC = %v\n", m.NumGC)
|
|
}
|
|
|
|
func bToMb(b uint64) uint64 {
|
|
return b / 1024 / 1024
|
|
}
|
|
|
|
func BenchmarkCount(b *testing.B) {
|
|
hm := New(HashInt, compare.Int)
|
|
b.N = 100000
|
|
for i := 0; i < b.N; i++ {
|
|
hm.Put(i, i)
|
|
}
|
|
|
|
b.Log(len(hm.table), hm.size)
|
|
PrintMemUsage()
|
|
}
|
|
|
|
func BenchmarkGoCount(b *testing.B) {
|
|
m := make(map[int]int)
|
|
b.N = 100000
|
|
for i := 0; i < b.N; i++ {
|
|
m[i] = i
|
|
}
|
|
b.Log(len(m))
|
|
PrintMemUsage()
|
|
}
|