Small update in benchmark code

zaf · Jan 21, 2017 · 38a15db · 38a15db
1 parent 515098d
commit 38a15db
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/agi_test.go b/agi_test.go
@@ -236,6 +236,8 @@ func TestParseRespomse(t *testing.T) {
 
 // Benchmark AGI session initialisation
 func BenchmarkParseEnv(b *testing.B) {
+	b.SetBytes(int64(len(env)))
+	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
 		a := New()
 		a.Init(
@@ -252,6 +254,7 @@ func BenchmarkParseRes(b *testing.B) {
 	read := make([]byte, 0, len(repVal)+len(rep))
 	read = append(read, repVal...)
 	read = append(read, rep...)
+	b.SetBytes(int64(len(read)))
 	a := New()
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {