Skip to content

Commit 71a9966

Browse files
committed
test: update test_iter_batches_dataframe to assert RecordBatch type and conversion
1 parent b5e1f5b commit 71a9966

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

python/tests/test_dataframe.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1624,7 +1624,8 @@ def test_iter_batches_dataframe(fail_collect):
16241624

16251625
expected = [batch1, batch2]
16261626
for got, exp in zip(df, expected):
1627-
assert got.equals(exp)
1627+
assert isinstance(got, RecordBatch)
1628+
assert got.to_pyarrow().equals(exp)
16281629

16291630

16301631
def test_arrow_c_stream_to_table_and_reader(fail_collect):

0 commit comments

Comments
 (0)