| 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 | }; |
no test coverage detected