| 711 | |
| 712 | protected: |
| 713 | void MakeBatchesAndReader(int length) { |
| 714 | auto field1 = field("f1", int32()); |
| 715 | auto field2 = field("f2", uint8()); |
| 716 | auto field3 = field("f3", int16()); |
| 717 | |
| 718 | auto schema = ::arrow::schema({field1, field2, field3}); |
| 719 | |
| 720 | random::RandomArrayGenerator gen(42); |
| 721 | |
| 722 | auto array1_1 = gen.ArrayOf(int32(), length); |
| 723 | auto array1_2 = gen.ArrayOf(int32(), length); |
| 724 | auto array1_3 = gen.ArrayOf(int32(), length); |
| 725 | |
| 726 | auto array2_1 = gen.ArrayOf(uint8(), length); |
| 727 | auto array2_2 = gen.ArrayOf(uint8(), length); |
| 728 | auto array2_3 = gen.ArrayOf(uint8(), length); |
| 729 | |
| 730 | auto array3_1 = gen.ArrayOf(int16(), length); |
| 731 | auto array3_2 = gen.ArrayOf(int16(), length); |
| 732 | auto array3_3 = gen.ArrayOf(int16(), length); |
| 733 | |
| 734 | auto batch1 = RecordBatch::Make(schema, length, {array1_1, array2_1, array3_1}); |
| 735 | auto batch2 = RecordBatch::Make(schema, length, {array1_2, array2_2, array3_2}); |
| 736 | auto batch3 = RecordBatch::Make(schema, length, {array1_3, array2_3, array3_3}); |
| 737 | |
| 738 | batches_ = {batch1, batch2, batch3}; |
| 739 | |
| 740 | ASSERT_OK_AND_ASSIGN(reader_, RecordBatchReader::Make(batches_)); |
| 741 | } |
| 742 | std::vector<std::shared_ptr<RecordBatch>> batches_; |
| 743 | std::shared_ptr<RecordBatchReader> reader_; |
| 744 | }; |
nothing calls this directly
no test coverage detected