MCPcopy Create free account
hub / github.com/apache/arrow / test_feather_format

Function test_feather_format

python/pyarrow/tests/test_dataset.py:3703–3732  ·  view source on GitHub ↗
(tempdir, dataset_reader)

Source from the content-addressed store, hash-verified

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

Callers

nothing calls this directly

Calls 6

write_featherFunction · 0.90
IpcFileFormatMethod · 0.80
equalsMethod · 0.80
arrayMethod · 0.45
to_tableMethod · 0.45

Tested by

no test coverage detected