From 91e1ef7f518d11bd5a9786b4cfb8a696d24626f4 Mon Sep 17 00:00:00 2001 From: Jorge Leitao Date: Tue, 12 Jul 2022 08:51:53 -0700 Subject: [PATCH] Added more tests (#1158) --- tests/it/io/parquet/mod.rs | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) diff --git a/tests/it/io/parquet/mod.rs b/tests/it/io/parquet/mod.rs index 6710bf57696..346208e8951 100644 --- a/tests/it/io/parquet/mod.rs +++ b/tests/it/io/parquet/mod.rs @@ -1394,3 +1394,38 @@ fn large_list_large_binary() -> Result<()> { array.try_extend(data).unwrap(); list_array_generic(false, array.into()) } + +#[test] +fn list_utf8_nullable() -> Result<()> { + let data = vec![ + Some(vec![Some("a".to_string())]), + None, + Some(vec![None, Some("b".to_string())]), + Some(vec![]), + Some(vec![Some("c".to_string())]), + None, + ]; + let mut array = + MutableListArray::::new_with_field(MutableUtf8Array::::new(), "item", true); + array.try_extend(data).unwrap(); + list_array_generic(true, array.into()) +} + +#[test] +fn list_int_nullable() -> Result<()> { + let data = vec![ + Some(vec![Some(1)]), + None, + Some(vec![None, Some(2)]), + Some(vec![]), + Some(vec![Some(3)]), + None, + ]; + let mut array = MutableListArray::::new_with_field( + MutablePrimitiveArray::::new(), + "item", + true, + ); + array.try_extend(data).unwrap(); + list_array_generic(true, array.into()) +}