Function test_struct_equal_null

arrow/tests/array_equal.rs:909–1009

()

#[test]
fn test_struct_equal_null() {
    let strings: ArrayRef = Arc::new(StringArray::from(vec![
        Some("joe"),
        None,
        None,
        Some("mark"),
        Some("doe"),
    ]));
    let ints: ArrayRef = Arc::new(Int32Array::from(vec![
        Some(1),
        Some(2),
        None,
        Some(4),
        Some(5),
    ]));
    let ints_non_null: ArrayRef = Arc::new(Int32Array::from(vec![1, 2, 3, 4, 0]));

    let a = ArrayData::builder(DataType::Struct(Fields::from(vec![
        Field::new("f1", DataType::Utf8, true),
        Field::new("f2", DataType::Int32, true),
    ])))
    .null_bit_buffer(Some(Buffer::from([0b00001011])))
    .len(5)
    .add_child_data(strings.to_data())
    .add_child_data(ints.to_data())
    .build()
    .unwrap();
    let a = make_array(a);

    let b = ArrayData::builder(DataType::Struct(Fields::from(vec![
        Field::new("f1", DataType::Utf8, true),
        Field::new("f2", DataType::Int32, true),
    ])))
    .null_bit_buffer(Some(Buffer::from([0b00001011])))
    .len(5)
    .add_child_data(strings.to_data())
    .add_child_data(ints_non_null.to_data())
    .build()
    .unwrap();
    let b = make_array(b);

    test_equal(&a, &b, true);

    // test with arrays that are not equal
    let c_ints_non_null: ArrayRef = Arc::new(Int32Array::from(vec![1, 2, 3, 0, 4]));
    let c = ArrayData::builder(DataType::Struct(Fields::from(vec![
        Field::new("f1", DataType::Utf8, true),
        Field::new("f2", DataType::Int32, true),
    ])))
    .null_bit_buffer(Some(Buffer::from([0b00001011])))
    .len(5)
    .add_child_data(strings.to_data())
    .add_child_data(c_ints_non_null.to_data())
    .build()
    .unwrap();
    let c = make_array(c);

    test_equal(&a, &c, false);