We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 50fe1c1 commit 1b96ddcCopy full SHA for 1b96ddc
1 file changed
python/tests/test_dataframe.py
@@ -1629,9 +1629,11 @@ def test_iter_batches_dataframe(fail_collect):
1629
df = ctx.create_dataframe([[batch1], [batch2]])
1630
1631
expected = [batch1, batch2]
1632
- for got, exp in zip(df, expected):
1633
- assert isinstance(got, RecordBatch)
1634
- assert got.to_pyarrow().equals(exp)
+ results = [b.to_pyarrow() for b in df]
+
+ assert len(results) == len(expected)
1635
+ for exp in expected:
1636
+ assert any(got.equals(exp) for got in results)
1637
1638
1639
def test_arrow_c_stream_to_table_and_reader(fail_collect):
0 commit comments