reduce code duplication in benchmark tests

1 files changed, 24 insertions, 0 deletions

diff --git a/internal/must/must.go b/internal/must/must.go
new file mode 100644
index 0000000..e472579
--- /dev/null
+++ b/internal/must/must.go
@@ -0,0 +1,24 @@
+// SPDX-FileCopyrightText: 2026 Stefan Majewsky <majewsky@gmx.net>
+// SPDX-License-Identifier: Apache-2.0
+
+package must
+
+import "testing"
+
+// Succeed fails the test if err is not nil.
+func Succeed(t testing.TB, err error) {
+	if err != nil {
+		t.Fatal(err.Error())
+	}
+}
+
+// Return wraps a function returning two output values,
+// and either forwards the result value on success, or fails the test on error.
+func Return[V any](value V, err error) func(testing.TB) V {
+	return func(t testing.TB) V {
+		if err != nil {
+			t.Fatal(err.Error())
+		}
+		return value
+	}
+}