(arrow-rs) branch master updated: add view type single column tests (#5891)

[alamb]

This is an automated email from the ASF dual-hosted git repository.

alamb pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/arrow-rs.git


The following commit(s) were added to refs/heads/master by this push:
     new cc0ce1b4796 add view type single column tests (#5891)
cc0ce1b4796 is described below

commit cc0ce1b4796969de0df76c814049d73c6bcabb78
Author: Yijun Zhao <ariesdevi...@gmail.com>
AuthorDate: Sun Jun 16 00:58:43 2024 +0800

    add view type single column tests (#5891)
---
 parquet/src/arrow/arrow_writer/mod.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/parquet/src/arrow/arrow_writer/mod.rs 
b/parquet/src/arrow/arrow_writer/mod.rs
index 53287dec572..9d48ea9e463 100644
--- a/parquet/src/arrow/arrow_writer/mod.rs
+++ b/parquet/src/arrow/arrow_writer/mod.rs
@@ -2103,6 +2103,16 @@ mod tests {
         values_required::<BinaryArray, _>(many_vecs_iter);
     }
 
+    #[test]
+    fn binary_view_single_column() {
+        let one_vec: Vec<u8> = (0..SMALL_SIZE as u8).collect();
+        let many_vecs: Vec<_> = 
std::iter::repeat(one_vec).take(SMALL_SIZE).collect();
+        let many_vecs_iter = many_vecs.iter().map(|v| v.as_slice());
+
+        // BinaryArrays can't be built from Vec<Option<&str>>, so only call 
`values_required`
+        values_required::<BinaryViewArray, _>(many_vecs_iter);
+    }
+
     #[test]
     fn i32_column_bloom_filter() {
         let array = Arc::new(Int32Array::from_iter(0..SMALL_SIZE as i32));
@@ -2195,6 +2205,14 @@ mod tests {
         required_and_optional::<LargeStringArray, _>(raw_strs);
     }
 
+    #[test]
+    fn string_view_single_column() {
+        let raw_values: Vec<_> = (0..SMALL_SIZE).map(|i| 
i.to_string()).collect();
+        let raw_strs = raw_values.iter().map(|s| s.as_str());
+
+        required_and_optional::<StringViewArray, _>(raw_strs);
+    }
+
     #[test]
     fn null_list_single_column() {
         let null_field = Field::new("item", DataType::Null, true);