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

Method CheckUnionExecNode

cpp/src/arrow/acero/union_node_test.cc:121–137  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

119 }
120
121 void CheckUnionExecNode(size_t num_input_nodes, size_t num_batches, bool parallel) {
122 auto random_schema = GenerateRandomSchema(num_input_nodes);
123
124 int multiplicity = parallel ? 10 : 1;
125 std::vector<std::shared_ptr<RecordBatch>> all_record_batches;
126 std::vector<BatchesWithSchema> input_batches(num_input_nodes);
127 BatchesWithSchema exp_batches;
128 exp_batches.schema = random_schema;
129 for (size_t i = 0; i < num_input_nodes; i++) {
130 GenerateBatchesFromSchema(random_schema, num_batches, &input_batches[i],
131 multiplicity, kBatchSize);
132 for (const auto& batch : input_batches[i].batches) {
133 exp_batches.batches.push_back(batch);
134 }
135 }
136 CheckRunOutput(input_batches, exp_batches, parallel);
137 }
138
139 ::arrow::random::RandomArrayGenerator rng_;
140};

Callers 1

TEST_FFunction · 0.80

Calls 3

push_backMethod · 0.80
CheckRunOutputFunction · 0.70

Tested by

no test coverage detected