diff --git a/barbench_test.go b/barbench_test.go index 361c5de..0ef4fea 100644 --- a/barbench_test.go +++ b/barbench_test.go @@ -5,38 +5,27 @@ "testing" ) -func benchmarkSingleBar(total int) { - p := New(WithOutput(ioutil.Discard)) - bar := p.AddBar(int64(total)) - for i := 0; i < total; i++ { - bar.Increment() - } - p.Wait() -} - -func BenchmarkSingleBar100(b *testing.B) { - for i := 0; i < b.N; i++ { - benchmarkSingleBar(100) - } -} - -func BenchmarkSingleBar1000(b *testing.B) { - for i := 0; i < b.N; i++ { - benchmarkSingleBar(1000) - } -} - -func BenchmarkSingleBar10000(b *testing.B) { - for i := 0; i < b.N; i++ { - benchmarkSingleBar(10000) - } -} - func BenchmarkIncrSingleBar(b *testing.B) { p := New(WithOutput(ioutil.Discard)) bar := p.AddBar(int64(b.N)) for i := 0; i < b.N; i++ { bar.Increment() } - p.Wait() } + +func BenchmarkIncrSingleBarStartBlock(b *testing.B) { + p := New(WithOutput(ioutil.Discard)) + bar := p.AddBar(int64(b.N)) + for i := 0; i < b.N; i++ { + bar.StartBlock() + bar.Increment() + } +} + +func BenchmarkIncrSingleBarWhileIsNotCompleted(b *testing.B) { + p := New(WithOutput(ioutil.Discard)) + bar := p.AddBar(int64(b.N)) + for !bar.Completed() { + bar.Increment() + } +}