Merge pull request #23 from scop/map-benchmarks

Add map preallocation benchmarks

prealloc_test.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"go/token"
 	"testing"
 
@@ -125,3 +126,34 @@ func BenchmarkSize200PreallocateCopy(b *testing.B) {
 		copy(init, existing)
 	}
 }
+
+func BenchmarkMap(b *testing.B) {
+	benchmarks := []struct {
+		size        int
+		preallocate bool
+	}{
+		{10, false},
+		{10, true},
+		{200, false},
+		{200, true},
+	}
+	var m map[int]int
+	for _, bm := range benchmarks {
+		no := ""
+		if !bm.preallocate {
+			no = "No"
+		}
+		b.Run(fmt.Sprintf("Size%d%sPreallocate", bm.size, no), func(b *testing.B) {
+			for i := 0; i < b.N; i++ {
+				if bm.preallocate {
+					m = make(map[int]int, bm.size)
+				} else {
+					m = make(map[int]int)
+				}
+				for j := 0; j < bm.size; j++ {
+					m[j] = j
+				}
+			}
+		})
+	}
+}