Fix Benchmark tests.

tristanmorgan · Jun 1, 2024 · 20de37f · 20de37f
1 parent 1b8ff35
commit 20de37f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/parser/execute_test.go b/parser/execute_test.go
@@ -67,11 +67,11 @@ func BenchmarkExecute(b *testing.B) {
 		b.Errorf("error opening program: err:")
 	}
 	buff := bufio.NewReader(sourceFile)
-	inputBuf := bufio.NewReader(strings.NewReader("80\n"))
-	outputBuf := bufio.NewWriter(&bufferWriter{})
 	program, _ := Tokenise(buff)
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
+		inputBuf := bufio.NewReader(strings.NewReader("80\n"))
+		outputBuf := bufio.NewWriter(&bufferWriter{})
 		startdata := make([]int, 65536)
 		Execute(startdata, program, inputBuf, outputBuf)
 	}

diff --git a/parser/tokenise_test.go b/parser/tokenise_test.go
@@ -157,9 +157,9 @@ func BenchmarkTokenise(b *testing.B) {
 	if err != nil {
 		b.Errorf("error opening program: err:")
 	}
-	buff := bufio.NewReader(sourceFile)
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
+		buff := bufio.NewReader(sourceFile)
 		Tokenise(buff)
 	}
 }