| 84 | } |
| 85 | |
| 86 | RecordBatchVector GetBatches(size_t num_batches, size_t batch_size) { |
| 87 | auto iter = datasets.find(std::make_pair(num_batches, batch_size)); |
| 88 | if (iter == datasets.end()) { |
| 89 | return RecordBatchVector{}; |
| 90 | } |
| 91 | return iter->second; |
| 92 | } |
| 93 | |
| 94 | std::shared_ptr<Schema> GetSchema() { |
| 95 | static std::shared_ptr<Schema> s = schema({field("a", int32()), field("b", boolean())}); |
no test coverage detected