Method WriteTestData

cpp/src/parquet/reader_test.cc:1957–1989

  const int kNumRows = 1000;

  void WriteTestData(std::shared_ptr<const LogicalType> type, bool write_arrow) {
    // Make schema
    schema::NodeVector fields;
    fields.push_back(
        PrimitiveNode::Make("g", Repetition::REQUIRED, type, Type::BYTE_ARRAY));
    auto schema = std::static_pointer_cast<GroupNode>(
        GroupNode::Make("schema", Repetition::REQUIRED, fields));

    // Write small batches and small data pages
    auto writer_props_builder = WriterProperties::Builder();
    writer_props_builder.write_batch_size(64);

    std::shared_ptr<WriterProperties> writer_props = writer_props_builder.build();

    ASSERT_OK_AND_ASSIGN(auto out_file, ::arrow::io::BufferOutputStream::Create());
    std::shared_ptr<ParquetFileWriter> file_writer =
        ParquetFileWriter::Open(out_file, schema, writer_props);
    RowGroupWriter* rg_writer = file_writer->AppendRowGroup();

    // write WKB points to columns
    auto* writer =
        ::arrow::internal::checked_cast<ByteArrayWriter*>(rg_writer->NextColumn());
    if (!write_arrow) {
      WriteTestDataUsingWriteBatch(writer);
    } else {
      WriteTestDataUsingWriteArrow(writer);
    }

    rg_writer->Close();
    file_writer->Close();

    ASSERT_OK_AND_ASSIGN(file_buf, out_file->Finish());
  }

  void WriteTestDataUsingWriteBatch(ByteArrayWriter* writer) {
    std::vector<uint8_t> buffer(test::kWkbPointXYSize * kNumRows);