diff --git a/parquet/src/util/bit_util.rs b/parquet/src/util/bit_util.rs index d0b13072dc4e..8599e1334294 100644 --- a/parquet/src/util/bit_util.rs +++ b/parquet/src/util/bit_util.rs @@ -1223,11 +1223,30 @@ mod tests { const SIZE: &[usize] = &[1, 31, 32, 33, 128, 129]; for s in SIZE { for i in 0..=64 { + // `get_batch` is generic over all of these types, so exercise the + // signed integer types and `bool` alongside the unsigned types rather + // than relying on the unsigned coverage alone. match i { - 0..=8 => test_get_batch_helper::(*s, i), - 9..=16 => test_get_batch_helper::(*s, i), - 17..=32 => test_get_batch_helper::(*s, i), - _ => test_get_batch_helper::(*s, i), + 0..=8 => { + test_get_batch_helper::(*s, i); + test_get_batch_helper::(*s, i); + } + 9..=16 => { + test_get_batch_helper::(*s, i); + test_get_batch_helper::(*s, i); + } + 17..=32 => { + test_get_batch_helper::(*s, i); + test_get_batch_helper::(*s, i); + } + _ => { + test_get_batch_helper::(*s, i); + test_get_batch_helper::(*s, i); + } + } + // `bool` only supports a bit width of 1. + if i == 1 { + test_get_batch_helper::(*s, i); } } }