(tempdir, dataset_reader)
| 3700 | |
| 3701 | |
| 3702 | def test_feather_format(tempdir, dataset_reader): |
| 3703 | from pyarrow.feather import write_feather |
| 3704 | |
| 3705 | table = pa.table({'a': pa.array([1, 2, 3], type="int8"), |
| 3706 | 'b': pa.array([.1, .2, .3], type="float64")}) |
| 3707 | |
| 3708 | basedir = tempdir / "feather_dataset" |
| 3709 | basedir.mkdir() |
| 3710 | write_feather(table, str(basedir / "data.feather")) |
| 3711 | |
| 3712 | dataset = ds.dataset(basedir, format=ds.IpcFileFormat()) |
| 3713 | result = dataset_reader.to_table(dataset) |
| 3714 | assert result.equals(table) |
| 3715 | |
| 3716 | assert_dataset_fragment_convenience_methods(dataset) |
| 3717 | |
| 3718 | dataset = ds.dataset(basedir, format="feather") |
| 3719 | result = dataset_reader.to_table(dataset) |
| 3720 | assert result.equals(table) |
| 3721 | |
| 3722 | # ARROW-8641 - column selection order |
| 3723 | result = dataset_reader.to_table(dataset, columns=["b", "a"]) |
| 3724 | assert result.column_names == ["b", "a"] |
| 3725 | result = dataset_reader.to_table(dataset, columns=["a", "a"]) |
| 3726 | assert result.column_names == ["a", "a"] |
| 3727 | |
| 3728 | # error with Feather v1 files |
| 3729 | write_feather(table, str(basedir / "data1.feather"), version=1) |
| 3730 | with pytest.raises(ValueError): |
| 3731 | dataset_reader.to_table(ds.dataset(basedir, format="feather")) |
| 3732 | |
| 3733 | |
| 3734 | @pytest.mark.pandas |
nothing calls this directly
no test coverage detected