Method test_flatten

c_glib/test/test-struct-array.rb:53–87

  end

  def test_flatten
    fields = [
      Arrow::Field.new("score", Arrow::Int8DataType.new),
      Arrow::Field.new("enabled", Arrow::BooleanDataType.new),
    ]
    data_type = Arrow::StructDataType.new(fields)
    builder = Arrow::StructArrayBuilder.new(data_type)

    builder.append_value
    builder.get_field_builder(0).append_value(-29)
    builder.get_field_builder(1).append_value(true)

    builder.append_value
    builder.field_builders[0].append_value(2)
    builder.field_builders[1].append_value(false)

    array = builder.finish
    values = array.length.times.collect do |i|
      if i.zero?
        [
          array.get_field(0).get_value(i),
          array.get_field(1).get_value(i),
        ]
      else
        array.flatten.collect do |field|
          field.get_value(i)
        end
      end
    end
    assert_equal([
                   [-29, true],
                   [2, false],
                 ],
                 values)
  end
end