diff --git a/crates/polars-parquet/src/parquet/encoding/byte_stream_split/mod.rs b/crates/polars-parquet/src/parquet/encoding/byte_stream_split/mod.rs index d84fa4820ab6..1ef6a99a2128 100644 --- a/crates/polars-parquet/src/parquet/encoding/byte_stream_split/mod.rs +++ b/crates/polars-parquet/src/parquet/encoding/byte_stream_split/mod.rs @@ -41,7 +41,7 @@ mod tests { } #[test] - fn fails_for_bad_size() -> Result<(), ParquetError> { + fn fails_for_invalid_values_size() -> Result<(), ParquetError> { let buffer = vec![0; 12]; let result = Decoder::try_new(&buffer, 8); @@ -50,6 +50,16 @@ mod tests { Ok(()) } + #[test] + fn fails_for_invalid_element_size() -> Result<(), ParquetError> { + let buffer = vec![0; 16]; + + let result = Decoder::try_new(&buffer, 16); + assert!(result.is_err()); + + Ok(()) + } + fn encode(data: &[T], buffer: &mut Vec) { let element_size = std::mem::size_of::(); let num_elements = data.len();