(num_cols=2, num_rows=10, linesep='\r\n', write_names=True)
| 66 | |
| 67 | |
| 68 | def make_random_csv(num_cols=2, num_rows=10, linesep='\r\n', write_names=True): |
| 69 | rnd = random.Random(42) |
| 70 | arr = [rnd.randint(0, 1000) for _ in range(num_cols * num_rows)] |
| 71 | csv = io.StringIO() |
| 72 | col_names = list(itertools.islice(generate_col_names(), num_cols)) |
| 73 | if write_names: |
| 74 | csv.write(",".join(col_names)) |
| 75 | csv.write(linesep) |
| 76 | for row in split_rows(arr, num_cols, num_rows): |
| 77 | csv.write(",".join(map(str, row))) |
| 78 | csv.write(linesep) |
| 79 | csv = csv.getvalue().encode() |
| 80 | columns = [pa.array(row, type=pa.int64()) |
| 81 | for row in split_columns(arr, num_cols, num_rows)] |
| 82 | expected = pa.Table.from_arrays(columns, col_names) |
| 83 | return csv, expected |
| 84 | |
| 85 | |
| 86 | def make_empty_csv(column_names): |
no test coverage detected