Function test_array_with_nulls

datafusion/functions/src/unicode/reverse.rs:239–290

()

    #[test]
    fn test_array_with_nulls() {
        use crate::unicode::reverse::reverse;
        use arrow::array::ArrayRef;
        use std::sync::Arc;

        let input_values = vec![Some("abcd"), None, Some("XYZ"), Some("héllo"), None];
        let expected: Vec<Option<&str>> =
            vec![Some("dcba"), None, Some("ZYX"), Some("olléh"), None];

        let cases: Vec<(&str, ArrayRef)> = vec![
            (
                "StringArray",
                Arc::new(StringArray::from(input_values.clone())),
            ),
            (
                "LargeStringArray",
                Arc::new(LargeStringArray::from(input_values.clone())),
            ),
            (
                "StringViewArray",
                Arc::new(StringViewArray::from(input_values.clone())),
            ),
        ];

        for (label, input) in cases {
            let out = reverse(&[input]).unwrap();
            assert_eq!(out.len(), expected.len(), "{label}: length mismatch");

            let actual: Vec<Option<&str>> = match out.data_type() {
                Utf8 => out
                    .as_any()
                    .downcast_ref::<StringArray>()
                    .unwrap()
                    .iter()
                    .collect(),
                LargeUtf8 => out
                    .as_any()
                    .downcast_ref::<LargeStringArray>()
                    .unwrap()
                    .iter()
                    .collect(),
                Utf8View => out
                    .as_any()
                    .downcast_ref::<StringViewArray>()
                    .unwrap()
                    .iter()
                    .collect(),
                other => panic!("{label}: unexpected output type {other:?}"),
            };
            assert_eq!(actual, expected, "{label}: value mismatch");
        }
    }
}