| 442 | } |
| 443 | |
| 444 | Result<std::shared_ptr<RecordBatch>> DoLargeRoundTrip(const RecordBatch& batch, |
| 445 | bool zero_data) { |
| 446 | if (zero_data) { |
| 447 | RETURN_NOT_OK(ZeroMemoryMap(mmap_.get())); |
| 448 | } |
| 449 | RETURN_NOT_OK(mmap_->Seek(0)); |
| 450 | |
| 451 | auto options = options_; |
| 452 | options.allow_64bit = true; |
| 453 | |
| 454 | ARROW_ASSIGN_OR_RAISE(auto file_writer, |
| 455 | MakeFileWriter(mmap_, batch.schema(), options)); |
| 456 | RETURN_NOT_OK(file_writer->WriteRecordBatch(batch)); |
| 457 | RETURN_NOT_OK(file_writer->Close()); |
| 458 | |
| 459 | ARROW_ASSIGN_OR_RAISE(int64_t offset, mmap_->Tell()); |
| 460 | |
| 461 | std::shared_ptr<RecordBatchFileReader> file_reader; |
| 462 | ARROW_ASSIGN_OR_RAISE(file_reader, RecordBatchFileReader::Open(mmap_.get(), offset)); |
| 463 | |
| 464 | return file_reader->ReadRecordBatch(0); |
| 465 | } |
| 466 | |
| 467 | void CheckReadResult(const RecordBatch& result, const RecordBatch& expected) { |
| 468 | ASSERT_OK(result.ValidateFull()); |
nothing calls this directly
no test coverage detected