test: FromIterator for VariableSizeListArray with nullable child …

…array
mbrobbel · Jul 26, 2023 · adc3f3b · adc3f3b
1 parent aeefe44
commit adc3f3b
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 1 deletion.
diff --git a/Cargo.toml b/Cargo.toml
@@ -15,7 +15,7 @@ members = ["narrow-derive"]
 
 [package]
 name = "narrow"
-version= "0.2.2"
+version= "0.2.3"
 authors.workspace = true
 edition.workspace = true
 rust-version.workspace = true

diff --git a/src/array/variable_size_list.rs b/src/array/variable_size_list.rs
@@ -164,5 +164,52 @@ mod tests {
         assert_eq!(array.0.data.0.data.is_valid(1), Some(true));
         assert_eq!(array.0.data.0.data.0.is_null(0), Some(true));
         assert_eq!(array.0.data.0.data.0.is_valid(1), Some(true));
+
+        let input = vec![
+            None,
+            Some(vec![
+                None,
+                Some(vec![
+                    None,
+                    Some(vec![None, Some(2), Some(3)]),
+                    Some(vec![None, Some(2), Some(3)]),
+                ]),
+                Some(vec![None, Some(vec![None, Some(5), Some(6)])]),
+            ]),
+        ];
+        let array = input.into_iter().collect::<VariableSizeListArray<
+            VariableSizeListArray<
+                VariableSizeListArray<FixedSizePrimitiveArray<u8, true>, true>,
+                true,
+            >,
+            true,
+        >>();
+        assert_eq!(
+            array.0.data.0.data.0.data.0.as_ref(),
+            &[
+                u8::default(),
+                2,
+                3,
+                u8::default(),
+                2,
+                3,
+                u8::default(),
+                5,
+                6
+            ]
+        );
+        assert_eq!(array.0.offsets.as_ref(), &[0, 0, 3]);
+        assert_eq!(array.0.data.0.offsets.as_ref(), &[0, 0, 3, 5]);
+        assert_eq!(array.0.data.0.data.0.offsets.as_ref(), &[0, 0, 3, 6, 6, 9]);
+        assert_eq!(array.is_null(0), Some(true));
+        assert_eq!(array.is_valid(1), Some(true));
+        assert_eq!(array.0.data.is_null(0), Some(true));
+        assert_eq!(array.0.data.is_valid(1), Some(true));
+        assert_eq!(array.0.data.0.data.is_null(0), Some(true));
+        assert_eq!(array.0.data.0.data.is_valid(1), Some(true));
+        assert_eq!(array.0.data.0.data.0.is_null(0), Some(true));
+        assert_eq!(array.0.data.0.data.0.is_valid(1), Some(true));
+        assert_eq!(array.0.data.0.data.0.data.0.is_null(0), Some(true));
+        assert_eq!(array.0.data.0.data.0.data.0.is_valid(1), Some(true));
     }
 }