diff --git a/x/staking/bench_test.go b/x/staking/bench_test.go index 3d83f81a47..4c80b9031a 100644 --- a/x/staking/bench_test.go +++ b/x/staking/bench_test.go @@ -39,8 +39,7 @@ func benchmarkValidateGenesis(b *testing.B, n int) { validators = append(validators, validator) } - b.ResetTimer() - for i := 0; i < b.N; i++ { + for b.Loop() { genesisState := types.DefaultGenesisState() genesisState.Validators = validators if err := staking.ValidateGenesis(genesisState); err != nil { diff --git a/x/staking/keeper_bench_test.go b/x/staking/keeper_bench_test.go index f7283f566b..760b6ee6ae 100644 --- a/x/staking/keeper_bench_test.go +++ b/x/staking/keeper_bench_test.go @@ -50,8 +50,8 @@ func BenchmarkApplyAndReturnValidatorSetUpdates(b *testing.B) { require.NoError(b, keeper.SetLastValidatorPower(ctx, valAddrs[i], validator.ConsensusPower(sdk.DefaultPowerReduction))) } ctx, _ = testEnv.ctx.CacheContext() - b.ResetTimer() - for i := 0; i < b.N; i++ { + + for b.Loop() { _, _ = keeper.ApplyAndReturnValidatorSetUpdates(ctx) } }