From af71181dd6aa639d2cd0380ca1f0cdb6b1512f32 Mon Sep 17 00:00:00 2001 From: vastonus Date: Thu, 16 Oct 2025 16:46:19 +0800 Subject: [PATCH] refactor: use b.Loop() in benchmark tests for better performance Signed-off-by: vastonus --- pkg/tss/elgamal_bench_test.go | 7 ++----- pkg/tss/schnorr_bench_test.go | 16 ++++------------ 2 files changed, 6 insertions(+), 17 deletions(-) diff --git a/pkg/tss/elgamal_bench_test.go b/pkg/tss/elgamal_bench_test.go index 0da34a18d..7f88bb38e 100644 --- a/pkg/tss/elgamal_bench_test.go +++ b/pkg/tss/elgamal_bench_test.go @@ -12,9 +12,8 @@ func BenchmarkEncrypt(b *testing.B) { value := testutil.HexDecode("fc93f14f4e3e4e15378e2c65ba1986494a3f54b7c135dd21d67a44435332eb71") key := testutil.HexDecode("035db2a125a23300bef24e57883f547503ab2598a99ed07d65d482b4ea1ff8ed26") rng := testutil.MockNonce16Generator{} - b.ResetTimer() - for i := 0; i < b.N; i++ { + for b.Loop() { if _, err := tss.Encrypt(value, key, rng); err != nil { b.Fatal(err) } @@ -27,9 +26,7 @@ func BenchmarkDecrypt(b *testing.B) { ) key := testutil.HexDecode("64540a84e00ca07eb2f34bfa98caf96c8db3b09918427bca2863ee0b2d6df31f") - b.ResetTimer() - - for i := 0; i < b.N; i++ { + for b.Loop() { if _, err := tss.Decrypt(enc, key); err != nil { b.Fatal(err) } diff --git a/pkg/tss/schnorr_bench_test.go b/pkg/tss/schnorr_bench_test.go index 1d007ab3a..f2da39edf 100644 --- a/pkg/tss/schnorr_bench_test.go +++ b/pkg/tss/schnorr_bench_test.go @@ -11,9 +11,7 @@ func BenchmarkSign(b *testing.B) { suite := new(TSSTestSuite) suite.SetupTest() - b.ResetTimer() - - for i := 0; i < b.N; i++ { + for b.Loop() { if _, err := tss.Sign(suite.privKey, suite.challenge, suite.nonce, nil); err != nil { b.Fatal(err) } @@ -28,9 +26,7 @@ func BenchmarkVerify(b *testing.B) { signatureR := signature.R() signatureS := signature.S() - b.ResetTimer() - - for i := 0; i < b.N; i++ { + for b.Loop() { if err := tss.Verify(signatureR, signatureS, suite.challenge, suite.pubKey, nil, nil); err != nil { b.Fatal(err) } @@ -48,9 +44,7 @@ func BenchmarkVerifyWithCustomGenerator(b *testing.B) { keySym, _ := tss.ComputeSecretSym(suite.privKey, generator) nonceSym, _ := tss.ComputeSecretSym(suite.nonce, generator) - b.ResetTimer() - - for i := 0; i < b.N; i++ { + for b.Loop() { if err := tss.Verify(nonceSym, signatureS, suite.challenge, keySym, generator, nil); err != nil { b.Fatal(err) } @@ -66,9 +60,7 @@ func BenchmarkVerifyWithCustomLagrange(b *testing.B) { signatureR := signature.R() signatureS := signature.S() - b.ResetTimer() - - for i := 0; i < b.N; i++ { + for b.Loop() { if err := tss.Verify(signatureR, signatureS, suite.challenge, suite.pubKey, nil, lagrange); err != nil { b.Fatal(err) }