| 44 | constexpr int kCsvCols = 10; |
| 45 | |
| 46 | std::shared_ptr<RecordBatch> MakeIntTestBatch(int rows, int cols, int64_t null_percent) { |
| 47 | random::RandomArrayGenerator rg(kSeed); |
| 48 | |
| 49 | FieldVector fields(cols); |
| 50 | ArrayVector arrays(cols); |
| 51 | for (int i = 0; i < cols; ++i) { |
| 52 | fields[i] = field('i' + std::to_string(i), int32()); |
| 53 | arrays[i] = rg.Int32(rows, -87654321, 123456789, null_percent / 100.); |
| 54 | } |
| 55 | return RecordBatch::Make(schema(fields), rows, arrays); |
| 56 | } |
| 57 | |
| 58 | std::shared_ptr<RecordBatch> MakeStrTestBatch(int rows, int cols, bool quote, |
| 59 | int64_t null_percent) { |