MCPcopy Create free account
hub / github.com/apache/arrow / MakeBatchesAndReader

Method MakeBatchesAndReader

cpp/src/arrow/record_batch_test.cc:713–741  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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};

Callers

nothing calls this directly

Calls 5

fieldFunction · 0.70
schemaFunction · 0.70
MakeFunction · 0.70
ASSERT_OK_AND_ASSIGNFunction · 0.70
ArrayOfMethod · 0.45

Tested by

no test coverage detected