@@ -217,11 +217,11 @@ impl NullBufferBuilder {
217
217
self . bitmap_builder . as_mut ( ) . map ( |b| b. as_slice_mut ( ) )
218
218
}
219
219
220
- /// Return the allocated size of this builder, in bits , useful for memory accounting.
220
+ /// Return the allocated size of this builder, in bytes , useful for memory accounting.
221
221
pub fn allocated_size ( & self ) -> usize {
222
222
self . bitmap_builder
223
223
. as_ref ( )
224
- . map ( |b| b. capacity ( ) )
224
+ . map ( |b| b. capacity ( ) / 8 )
225
225
. unwrap_or ( 0 )
226
226
}
227
227
}
@@ -250,7 +250,7 @@ mod tests {
250
250
builder. append_n_nulls ( 2 ) ;
251
251
builder. append_n_non_nulls ( 2 ) ;
252
252
assert_eq ! ( 6 , builder. len( ) ) ;
253
- assert_eq ! ( 512 , builder. allocated_size( ) ) ;
253
+ assert_eq ! ( 64 , builder. allocated_size( ) ) ;
254
254
255
255
let buf = builder. finish ( ) . unwrap ( ) ;
256
256
assert_eq ! ( & [ 0b110010_u8 ] , buf. validity( ) ) ;
@@ -263,7 +263,7 @@ mod tests {
263
263
builder. append_n_nulls ( 2 ) ;
264
264
builder. append_slice ( & [ false , false , false ] ) ;
265
265
assert_eq ! ( 6 , builder. len( ) ) ;
266
- assert_eq ! ( 512 , builder. allocated_size( ) ) ;
266
+ assert_eq ! ( 64 , builder. allocated_size( ) ) ;
267
267
268
268
let buf = builder. finish ( ) . unwrap ( ) ;
269
269
assert_eq ! ( & [ 0b0_u8 ] , buf. validity( ) ) ;
@@ -327,7 +327,7 @@ mod tests {
327
327
builder. append_null ( ) ;
328
328
builder. append_non_null ( ) ;
329
329
assert_eq ! ( builder. as_slice( ) . unwrap( ) , & [ 0xFF , 0b10111111 ] ) ;
330
- assert_eq ! ( builder. allocated_size( ) , 512 ) ;
330
+ assert_eq ! ( builder. allocated_size( ) , 64 ) ;
331
331
}
332
332
333
333
#[ test]
0 commit comments