Skip to content

Commit

Permalink
feather -> ipc in test
Browse files Browse the repository at this point in the history
  • Loading branch information
jorisvandenbossche authored and pitrou committed Nov 18, 2024
1 parent 1961a14 commit d506361
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions python/pyarrow/tests/test_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -4683,22 +4683,20 @@ def test_write_iterable(tempdir):
base_dir = tempdir / 'inmemory_iterable'
ds.write_dataset((batch for batch in table.to_batches()), base_dir,
schema=table.schema,
basename_template='dat_{i}.arrow', format="feather")
basename_template='dat_{i}.arrow', format="ipc")
result = ds.dataset(base_dir, format="ipc").to_table()
assert result.equals(table)

base_dir = tempdir / 'inmemory_reader'
reader = pa.RecordBatchReader.from_batches(table.schema,
table.to_batches())
ds.write_dataset(reader, base_dir,
basename_template='dat_{i}.arrow', format="feather")
ds.write_dataset(reader, base_dir, basename_template='dat_{i}.arrow', format="ipc")
result = ds.dataset(base_dir, format="ipc").to_table()
assert result.equals(table)

base_dir = tempdir / 'inmemory_pycapsule'
stream = TableStreamWrapper(table)
ds.write_dataset(stream, base_dir,
basename_template='dat_{i}.arrow', format="feather")
ds.write_dataset(stream, base_dir, basename_template='dat_{i}.arrow', format="ipc")
result = ds.dataset(base_dir, format="ipc").to_table()
assert result.equals(table)

Expand Down

0 comments on commit d506361

Please sign in to comment.