Method test_write_table

c_glib/test/test-csv-writer.rb:59–95

  end

  def test_write_table
    message_data = ["Start", "Shutdown", "Reboot"]
    count_data = [2, 9, 5]
    message_field = Arrow::Field.new("message", Arrow::StringDataType.new)
    count_field = Arrow::Field.new("count", Arrow::Int64DataType.new)
    schema = Arrow::Schema.new([message_field, count_field])

    buffer = Arrow::ResizableBuffer.new(0)
    output = Arrow::BufferOutputStream.new(buffer)
    begin
      csv_writer = Arrow::CSVWriter.new(output, schema)
      begin
        table = Arrow::Table.new(schema,
                                 [
                                   build_string_array(message_data),
                                   build_int64_array(count_data),
                                 ])
        csv_writer.write_table(table)
      ensure
        csv_writer.close
        assert do
          csv_writer.closed?
        end
      end
    ensure
      output.close
    end

    csv_output = buffer.data.to_s
    expected = <<~CSV
      "message","count"
      "Start",2
      "Shutdown",9
      "Reboot",5
    CSV
    assert_equal(expected, csv_output)
  end


  sub_test_case("options") do