test(encoding): cover inline bitpacking value limit

Author: u70b3

rust/lance-encoding/src/encodings/physical/bitpacking.rs

@@ -609,11 +609,11 @@ mod test {
         assert_eq!(decoded_values.as_ref(), values);
     }
 
-    fn assert_corrupt_unchunk<T>(data: LanceBuffer, expected_message: &str)
+    fn assert_corrupt_unchunk<T>(data: LanceBuffer, num_values: u64, expected_message: &str)
     where
         T: ArrowNativeType + BitPacking + Pod,
     {
-        let err = InlineBitpacking::unchunk::<T>(data, 1).unwrap_err();
+        let err = InlineBitpacking::unchunk::<T>(data, num_values).unwrap_err();
         assert!(matches!(err, lance_core::Error::CorruptFile { .. }));
         let err = err.to_string();
         assert!(
@@ -635,12 +635,21 @@ mod test {
     }
 
     #[rstest]
-    #[case::too_small_header(LanceBuffer::from(vec![1, 2, 3]), "too small")]
-    #[case::misaligned_chunk_size(LanceBuffer::from(vec![0, 0, 0, 0, 0]), "multiple")]
-    #[case::payload_size_mismatch(LanceBuffer::reinterpret_vec(vec![12_u32]), "payload")]
-    #[case::invalid_bit_width(LanceBuffer::reinterpret_vec(vec![33_u32]), "exceeds")]
-    fn unchunk_rejects(#[case] data: LanceBuffer, #[case] expected_message: &str) {
-        assert_corrupt_unchunk::<u32>(data, expected_message);
+    #[case::too_small_header(LanceBuffer::from(vec![1, 2, 3]), 1, "too small")]
+    #[case::misaligned_chunk_size(LanceBuffer::from(vec![0, 0, 0, 0, 0]), 1, "multiple")]
+    #[case::too_many_values(
+        LanceBuffer::reinterpret_vec(vec![0_u32]),
+        ELEMS_PER_CHUNK + 1,
+        "expected at most"
+    )]
+    #[case::payload_size_mismatch(LanceBuffer::reinterpret_vec(vec![12_u32]), 1, "payload")]
+    #[case::invalid_bit_width(LanceBuffer::reinterpret_vec(vec![33_u32]), 1, "exceeds")]
+    fn unchunk_rejects(
+        #[case] data: LanceBuffer,
+        #[case] num_values: u64,
+        #[case] expected_message: &str,
+    ) {
+        assert_corrupt_unchunk::<u32>(data, num_values, expected_message);
     }
 
     #[test_log::test(tokio::test)]