| 3008 | } |
| 3009 | |
| 3010 | std::shared_ptr<Buffer> MakeBooleanInt32Int64File(int num_rows, int num_batches) { |
| 3011 | auto schema_ = MakeBooleanInt32Int64Schema(); |
| 3012 | EXPECT_OK_AND_ASSIGN(auto sink, io::BufferOutputStream::Create(0)); |
| 3013 | EXPECT_OK_AND_ASSIGN(auto writer, MakeFileWriter(sink.get(), schema_)); |
| 3014 | |
| 3015 | std::shared_ptr<RecordBatch> batch; |
| 3016 | for (int i = 0; i < num_batches; i++) { |
| 3017 | ARROW_EXPECT_OK(MakeBooleanInt32Int64Batch(num_rows, &batch)); |
| 3018 | ARROW_EXPECT_OK(writer->WriteRecordBatch(*batch)); |
| 3019 | } |
| 3020 | |
| 3021 | ARROW_EXPECT_OK(writer->Close()); |
| 3022 | EXPECT_OK_AND_ASSIGN(auto buffer, sink->Finish()); |
| 3023 | return buffer; |
| 3024 | } |
| 3025 | |
| 3026 | void GetReadRecordBatchReadRanges( |
| 3027 | uint32_t num_rows, const std::vector<int>& included_fields, |
no test coverage detected