Skip to content

Commit 759fb86

Browse files
committed
Add alternative method for creating RecordBatchReader from Arrow C stream
1 parent 98ac3a1 commit 759fb86

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/tests/test_dataframe.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1654,6 +1654,8 @@ def test_arrow_c_stream_to_table_and_reader(fail_collect):
16541654
assert table.column("a").num_chunks == 2
16551655

16561656
reader = pa.RecordBatchReader._import_from_c_capsule(df.__arrow_c_stream__())
1657+
# alternatively, we could use
1658+
# reader = pa.RecordBatchReader.from_stream(df)
16571659
assert isinstance(reader, pa.RecordBatchReader)
16581660
reader_table = pa.Table.from_batches(reader)
16591661
expected = pa.Table.from_batches([batch1, batch2])

0 commit comments

Comments
 (0)