Function nullif_struct

datafusion/functions/src/core/nullif.rs:264–306

()

    #[test]
    fn nullif_struct() -> Result<()> {
        let fields = Fields::from(vec![
            Field::new("a", DataType::Int64, true),
            Field::new("b", DataType::Utf8, true),
        ]);

        let lhs_a = Arc::new(Int64Array::from(vec![Some(1), Some(2), None]));
        let lhs_b = Arc::new(StringArray::from(vec![Some("1"), Some("2"), None]));
        let lhs_nulls = Some(NullBuffer::from(vec![true, true, false]));
        let lhs = ColumnarValue::Array(Arc::new(StructArray::new(
            fields.clone(),
            vec![lhs_a, lhs_b],
            lhs_nulls,
        )));

        let rhs_a = Arc::new(Int64Array::from(vec![Some(1), Some(9), None]));
        let rhs_b = Arc::new(StringArray::from(vec![Some("1"), Some("2"), None]));
        let rhs_nulls = Some(NullBuffer::from(vec![true, true, false]));
        let rhs = ColumnarValue::Array(Arc::new(StructArray::new(
            fields.clone(),
            vec![rhs_a, rhs_b],
            rhs_nulls,
        )));

        let result = nullif_func(&[lhs, rhs])?;
        let result = result.into_array(0).expect("Failed to convert to array");

        let expected_arrays = vec![
            Arc::new(Int64Array::from(vec![None, Some(2), None])) as ArrayRef,
            Arc::new(StringArray::from(vec![None, Some("2"), None])) as ArrayRef,
        ];
        let expected_nulls = NullBuffer::from(vec![false, true, false]);

        let expected = Arc::new(StructArray::try_new(
            fields,
            expected_arrays,
            Some(expected_nulls),
        )?) as ArrayRef;

        assert_eq!(expected.as_ref(), result.as_ref());

        Ok(())
    }

    #[test]
    fn nullif_list() -> Result<()> {