Class StructArrayBuilderTest

ruby/red-arrow/test/test-struct-array-builder.rb:18–184

# under the License.

class StructArrayBuilderTest < Test::Unit::TestCase
  def setup
    @data_type = Arrow::StructDataType.new(visible: {type: :boolean},
                                           count: {type: :uint64})
    @builder = Arrow::StructArrayBuilder.new(@data_type)
  end

  sub_test_case("#append_value") do
    test("nil") do
      @builder.append_value(nil)
      array = @builder.finish
      assert_equal([
                     [false],
                     [0],
                   ],
                   [
                     array.find_field(0).to_a,
                     array.find_field(1).to_a,
                   ])
    end

    test("Array") do
      @builder.append_value([true, 1])
      @builder.append_value([])
      @builder.append_value([false])
      array = @builder.finish
      assert_equal([
                     [true, nil, false],
                     [1, nil, nil],
                   ],
                   [
                     array.find_field(0).to_a,
                     array.find_field(1).to_a,
                   ])
    end

    test("Arrow::Struct") do
      source_array = Arrow::StructArray.new(@data_type, [[true, 1]])
      struct = source_array.get_value(0)
      @builder.append_value(struct)
      array = @builder.finish
      assert_equal([
                     [true],
                     [1],
                   ],
                   [
                     array.find_field(0).to_a,
                     array.find_field(1).to_a,
                   ])
    end

    test("Hash") do
      @builder.append_value(count: 1, visible: true)
      @builder.append_value(visible: false)
      @builder.append_value(count: 2)
      array = @builder.finish
      assert_equal([
                     [true, false, nil],