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