diff options
Diffstat (limited to 'nizk/bench_test.go')
| -rw-r--r-- | nizk/bench_test.go | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/nizk/bench_test.go b/nizk/bench_test.go index f823394..9d1cc4b 100644 --- a/nizk/bench_test.go +++ b/nizk/bench_test.go @@ -1,7 +1,6 @@ package nizk import ( - "log" "math/rand" "slices" "sync" @@ -96,7 +95,7 @@ func BenchmarkFromPaper(b *testing.B) { } } -func runSeal(n int, bitlength int) { +func runSeal(n int, bitlength int, tb testing.TB) { var vals = make([]int, n) var ids = make([]Bytes, n) for i := range n { @@ -135,7 +134,7 @@ func runSeal(n int, bitlength int) { go func() { r[i], p1[i] = bits[i][idx].RevealStage1(Xs...) if !bits[i][idx].Commitment.VerifyStage1(c[i], r[i], p1[i]) { - log.Fatalf("bits[%d][%d] commitment failed to verify in stage1", i, idx) + tb.Fatalf("bits[%d][%d] commitment failed to verify in stage1", i, idx) } Zs[i] = r[i].Z wg.Done() @@ -165,7 +164,7 @@ func runSeal(n int, bitlength int) { go func() { r[i], p2[i] = bits[i][idx].RevealStage2(lost[i], bj[i], Xs...) if !bits[i][idx].Commitment.VerifyStage2(bj[i].StageCommitment, c[i], bj[i].StageReveal, r[i], p2[i]) { - log.Fatalf("bits[%d][%d] commitment failed to verify in stage2, result so far: %05b", i, idx, result) + tb.Fatalf("bits[%d][%d] commitment failed to verify in stage2, result so far: %05b", i, idx, result) } Zs[i] = r[i].Z wg.Done() @@ -187,13 +186,13 @@ func runSeal(n int, bitlength int) { } } if result != max { - log.Fatalf("wrong result: %0[1]*[2]b, expected: %0[1]*[3]b", bitlength, result, max) + tb.Fatalf("wrong result: %0[1]*[2]b, expected: %0[1]*[3]b", bitlength, result, max) } } func benchmarkMulti(n int, bitlength int, b *testing.B) { for range b.N { - runSeal(n, bitlength) + runSeal(n, bitlength, b) } } |