Skip to content

Commit 9e121db

Browse files
authored
CI: pyarrow nightly failures (#57005)
use RecordBatchReader.read_all to get table
1 parent 5740667 commit 9e121db

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pandas/tests/frame/test_arrow_interface.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,11 +35,11 @@ def test_dataframe_arrow_interface():
3535
def test_dataframe_to_arrow():
3636
df = pd.DataFrame({"a": [1, 2, 3], "b": ["a", "b", "c"]})
3737

38-
table = pa.RecordBatchReader.from_stream(df)
38+
table = pa.RecordBatchReader.from_stream(df).read_all()
3939
expected = pa.table({"a": [1, 2, 3], "b": ["a", "b", "c"]})
4040
assert table.equals(expected)
4141

4242
schema = pa.schema([("a", pa.int8()), ("b", pa.string())])
43-
table = pa.RecordBatchReader.from_stream(df, schema=schema)
43+
table = pa.RecordBatchReader.from_stream(df, schema=schema).read_all()
4444
expected = expected.cast(schema)
4545
assert table.equals(expected)

0 commit comments

Comments
 (0)