| 36 | using RecordBatchVector = std::vector<std::shared_ptr<RecordBatch>>; |
| 37 | |
| 38 | TEST_F(TestInMemoryFragment, Scan) { |
| 39 | constexpr int64_t kBatchSize = 1024; |
| 40 | constexpr int64_t kNumberBatches = 16; |
| 41 | |
| 42 | SetSchema({field("i32", int32()), field("f64", float64())}); |
| 43 | auto batch = ConstantArrayGenerator::Zeroes(kBatchSize, schema_); |
| 44 | auto reader = ConstantArrayGenerator::Repeat(kNumberBatches, batch); |
| 45 | |
| 46 | // Creates a InMemoryFragment of the same repeated batch. |
| 47 | RecordBatchVector batches = {static_cast<size_t>(kNumberBatches), batch}; |
| 48 | auto fragment = std::make_shared<InMemoryFragment>(batches); |
| 49 | |
| 50 | AssertFragmentEquals(reader.get(), fragment.get()); |
| 51 | } |
| 52 | |
| 53 | class TestInMemoryDataset : public DatasetFixtureMixin {}; |
| 54 |
nothing calls this directly
no test coverage detected