assertEqual: switch to testing.TB

myaaaaaaaaa · Dec 17, 2024 · 44f4048 · 44f4048
1 parent 6bf919c
commit 44f4048
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/gojqx_test.go b/gojqx_test.go
@@ -89,9 +89,7 @@ func BenchmarkShuffle(b *testing.B) {
 			`)
 
 			ss := slices.Collect(query(b.N))
-			if len(ss) != 1 {
-				panic(len(ss))
-			}
+			assertEqual(b, len(ss), 1)
 
 			buckets := ss[0].(map[string]any)
 			for k, v := range buckets {

diff --git a/util_test.go b/util_test.go
@@ -5,11 +5,11 @@ import (
 	"testing"
 )
 
-func assertEqual[T comparable](t *testing.T, got T, want T) {
-	t.Helper()
+func assertEqual[T comparable](tb testing.TB, got T, want T) {
+	tb.Helper()
 	if got != want {
-		t.Error("got", got)
-		t.Error("want", want)
+		tb.Error("got", got)
+		tb.Error("want", want)
 	}
 }
 func assertString(t *testing.T, got any, want string) {