Backport PR pandas-dev#57005: CI: pyarrow nightly failures

meeseeksmachine · Jan 22, 2024 · 317befa · 317befa
1 parent bfe6c4f
commit 317befa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pandas/tests/frame/test_arrow_interface.py b/pandas/tests/frame/test_arrow_interface.py
@@ -35,11 +35,11 @@ def test_dataframe_arrow_interface():
 def test_dataframe_to_arrow():
     df = pd.DataFrame({"a": [1, 2, 3], "b": ["a", "b", "c"]})
 
-    table = pa.RecordBatchReader.from_stream(df)
+    table = pa.RecordBatchReader.from_stream(df).read_all()
     expected = pa.table({"a": [1, 2, 3], "b": ["a", "b", "c"]})
     assert table.equals(expected)
 
     schema = pa.schema([("a", pa.int8()), ("b", pa.string())])
-    table = pa.RecordBatchReader.from_stream(df, schema=schema)
+    table = pa.RecordBatchReader.from_stream(df, schema=schema).read_all()
     expected = expected.cast(schema)
     assert table.equals(expected)