| 4864 | |
| 4865 | |
| 4866 | def test_write_scanner(tempdir, dataset_reader): |
| 4867 | table = pa.table([ |
| 4868 | pa.array(range(20)), pa.array(random.random() for _ in range(20)), |
| 4869 | pa.array(['a'] * 10 + ['b'] * 10) |
| 4870 | ], names=["f1", "f2", "part"]) |
| 4871 | dataset = ds.dataset(table) |
| 4872 | |
| 4873 | base_dir = tempdir / 'dataset_from_scanner' |
| 4874 | ds.write_dataset(dataset_reader.scanner( |
| 4875 | dataset), base_dir, format="feather") |
| 4876 | result = dataset_reader.to_table(ds.dataset(base_dir, format="ipc")) |
| 4877 | assert result.equals(table) |
| 4878 | |
| 4879 | # scanner with different projected_schema |
| 4880 | base_dir = tempdir / 'dataset_from_scanner2' |
| 4881 | ds.write_dataset(dataset_reader.scanner(dataset, columns=["f1"]), |
| 4882 | base_dir, format="feather") |
| 4883 | result = dataset_reader.to_table(ds.dataset(base_dir, format="ipc")) |
| 4884 | assert result.equals(table.select(["f1"])) |
| 4885 | |
| 4886 | # schema not allowed when writing a scanner |
| 4887 | with pytest.raises(ValueError, match="Cannot specify a schema"): |
| 4888 | ds.write_dataset(dataset_reader.scanner(dataset), base_dir, |
| 4889 | schema=table.schema, format="feather") |
| 4890 | |
| 4891 | |
| 4892 | def test_write_table_partitioned_dict(tempdir): |