improve the test consistency

Author: diegoholiveira

benchmark/bench.py

@@ -86,7 +86,7 @@ def run_benchmark(directory: str, is_published: bool, version: str) -> Dict[str,
 
     print(f"{Fore.YELLOW}Running benchmark for {version_name} version...{Style.RESET_ALL}")
     try:
-        result = subprocess.run(["go", "test", "-bench=.", "-benchmem", "-count=5"],
+        result = subprocess.run(["go", "test", "-bench=.", "-benchmem", "-count=10"],
                                cwd=directory, check=True, capture_output=True, text=True)
     except subprocess.CalledProcessError as e:
         print(f"{Fore.RED}Error running benchmark: {e}{Style.RESET_ALL}")

benchmark/benchmark_test.go

@@ -3,6 +3,7 @@ package benchmark
 import (
 	"bytes"
 	"fmt"
+	"runtime"
 	"strings"
 	"testing"
 
@@ -50,12 +51,33 @@ var TestCases = []struct {
 	},
 }
 
+func performWarmupRuns() {
+	runtime.GC()
+
+	for _, tc := range TestCases {
+		for i := 0; i < 10; i++ {
+			logic := strings.NewReader(tc.logic)
+			data := strings.NewReader(tc.data)
+			var result bytes.Buffer
+			_ = jsonlogic.Apply(logic, data, &result)
+		}
+	}
+
+	runtime.GC()
+}
+
 func BenchmarkJSONLogic(b *testing.B) {
+	performWarmupRuns()
+
 	for _, tc := range TestCases {
 		b.Run(tc.name, func(b *testing.B) {
 			b.ResetTimer()
 			b.ReportAllocs()
 
+			b.StopTimer()
+			runtime.GC()
+			b.StartTimer()
+
 			for i := 0; i < b.N; i++ {
 				logic := strings.NewReader(tc.logic)
 				data := strings.NewReader(tc.data)