| 470 | } |
| 471 | |
| 472 | std::shared_ptr<RecordBatch> MakeTestBatch(int idx) { |
| 473 | ArrayVector arrays; |
| 474 | // Row number |
| 475 | arrays.push_back(ArrayFromRange<Int32Type>(idx * kRowsPerTestBatch, |
| 476 | (idx + 1) * kRowsPerTestBatch, |
| 477 | /*add_nulls=*/false)); |
| 478 | // Filterable |
| 479 | arrays.push_back(ArrayFromRange<Int16Type>(0, kRowsPerTestBatch, |
| 480 | /*add_nulls=*/true)); |
| 481 | // Nested |
| 482 | std::shared_ptr<Array> x_vals = |
| 483 | ArrayFromRange<Int32Type>(0, kRowsPerTestBatch, /*add_nulls=*/false); |
| 484 | std::shared_ptr<Array> y_vals = |
| 485 | ArrayFromRange<Int32Type>(0, kRowsPerTestBatch, /*add_nulls=*/true); |
| 486 | EXPECT_OK_AND_ASSIGN(std::shared_ptr<Array> nested_arr, |
| 487 | StructArray::Make({std::move(x_vals), std::move(y_vals)}, |
| 488 | {field("x", int32()), field("y", int32())})); |
| 489 | arrays.push_back(std::move(nested_arr)); |
| 490 | return RecordBatch::Make(ScannerTestSchema(), kRowsPerTestBatch, std::move(arrays)); |
| 491 | } |
| 492 | |
| 493 | std::unique_ptr<MockDataset> MakeTestDataset(int num_fragments, int batches_per_fragment, |
| 494 | bool empty = false) { |
no test coverage detected