Function invoke_and_assert

datafusion/functions-nested/src/array_has.rs:1209–1234

Invoke a two-argument list UDF with the given arrays and assert the boolean output matches `expected`.

(
        udf: &dyn ScalarUDFImpl,
        haystack: &ArrayRef,
        needle: ArrayRef,
        expected: &[Option<bool>],
    )

    /// Invoke a two-argument list UDF with the given arrays and assert the
    /// boolean output matches `expected`.
    fn invoke_and_assert(
        udf: &dyn ScalarUDFImpl,
        haystack: &ArrayRef,
        needle: ArrayRef,
        expected: &[Option<bool>],
    ) {
        let num_rows = haystack.len();
        let list_type = haystack.data_type();
        let result = udf
            .invoke_with_args(ScalarFunctionArgs {
                args: vec![
                    ColumnarValue::Array(Arc::clone(haystack)),
                    ColumnarValue::Array(needle),
                ],
                arg_fields: vec![
                    Arc::new(Field::new("haystack", list_type.clone(), false)),
                    Arc::new(Field::new("needle", list_type.clone(), false)),
                ],
                number_rows: num_rows,
                return_field: Arc::new(Field::new("return", DataType::Boolean, true)),
                config_options: Arc::new(ConfigOptions::default()),
            })
            .unwrap();
        let output = result.into_array(num_rows).unwrap();
        assert_eq!(output.as_boolean().iter().collect::<Vec<_>>(), expected);
    }

    #[test]
    fn test_sliced_list_offsets() {