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

Method MakeBatchesAndReader

cpp/src/arrow/record_batch_test.cc:717–745  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

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