Function test_struct_array_builder

arrow-array/src/builder/struct_builder.rs:331–401

()

    #[test]
    fn test_struct_array_builder() {
        let string_builder = StringBuilder::new();
        let int_builder = Int32Builder::new();

        let fields = vec![
            Field::new("f1", DataType::Utf8, true),
            Field::new("f2", DataType::Int32, true),
        ];
        let field_builders = vec![
            Box::new(string_builder) as Box<dyn ArrayBuilder>,
            Box::new(int_builder) as Box<dyn ArrayBuilder>,
        ];

        let mut builder = StructBuilder::new(fields, field_builders);
        assert_eq!(2, builder.num_fields());

        let string_builder = builder
            .field_builder::<StringBuilder>(0)
            .expect("builder at field 0 should be string builder");
        string_builder.append_value("joe");
        string_builder.append_null();
        string_builder.append_null();
        string_builder.append_value("mark");
        string_builder.append_nulls(2);
        string_builder.append_value("terry");

        let int_builder = builder
            .field_builder::<Int32Builder>(1)
            .expect("builder at field 1 should be int builder");
        int_builder.append_value(1);
        int_builder.append_value(2);
        int_builder.append_null();
        int_builder.append_value(4);
        int_builder.append_nulls(2);
        int_builder.append_value(3);

        builder.append(true);
        builder.append(true);
        builder.append_null();
        builder.append(true);

        builder.append_nulls(2);
        builder.append(true);

        let struct_data = builder.finish().into_data();

        assert_eq!(7, struct_data.len());
        assert_eq!(3, struct_data.null_count());
        assert_eq!(&[75_u8], struct_data.nulls().unwrap().validity());

        let expected_string_data = ArrayData::builder(DataType::Utf8)
            .len(7)
            .null_bit_buffer(Some(Buffer::from(&[73_u8])))
            .add_buffer(Buffer::from_slice_ref([0, 3, 3, 3, 7, 7, 7, 12]))
            .add_buffer(Buffer::from_slice_ref(b"joemarkterry"))
            .build()
            .unwrap();