add benchmark for bitWriter WriteVlqInt()

apache · Apr 12, 2024 · 68cdc29 · 68cdc29
1 parent 1859298
commit 68cdc29
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/go/parquet/internal/utils/bit_reader_test.go b/go/parquet/internal/utils/bit_reader_test.go
@@ -70,6 +70,14 @@ func TestBitWriter(t *testing.T) {
 	assert.Equal(t, byte(2), buf[4])
 }
 
+func BenchmarkBitWriter(b *testing.B) {
+	buf := make([]byte, b.N)
+	bw := utils.NewBitWriter(utils.NewWriterAtBuffer(buf))
+	for i := 0; i < b.N; i++ {
+		assert.True(b, bw.WriteVlqInt(uint64(1)))
+	}
+}
+
 func TestBitReader(t *testing.T) {
 	buf := []byte{0xAA, 0xCC} // 0b10101010 0b11001100