Additional test

apache · May 8, 2024 · 2064a38 · 2064a38
1 parent a69cbac
commit 2064a38
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/arrow-array/src/array/union_array.rs b/arrow-array/src/array/union_array.rs
@@ -1366,6 +1366,14 @@ mod tests {
         UnionArray::try_new(fields.clone(), type_ids, None, children.clone()).unwrap();
 
         let type_ids = vec![1, 2].into();
+        let err =
+            UnionArray::try_new(fields.clone(), type_ids, None, children.clone()).unwrap_err();
+        assert_eq!(
+            err.to_string(),
+            "Invalid argument error: Type Ids values must match one of the field type ids"
+        );
+
+        let type_ids = vec![7, 2].into();
         let err = UnionArray::try_new(fields.clone(), type_ids, None, children).unwrap_err();
         assert_eq!(
             err.to_string(),