fix: improve DataFrame batch iteration tests for accuracy and clarity

Author: kosiew

python/tests/test_dataframe.py

@@ -1574,9 +1574,11 @@ def test_iter_batches_dataframe(fail_collect):
     batch2 = pa.record_batch([pa.array([2])], names=["a"])
     df = ctx.create_dataframe([[batch1], [batch2]])
 
-    expected = [batch1, batch2]
-    for got, exp in zip(df, expected):
-        assert got.equals(exp)
+    batches = list(df)
+
+    assert len(batches) == 2
+    assert any(batch.equals(batch1) for batch in batches)
+    assert any(batch.equals(batch2) for batch in batches)
 
 
 def test_arrow_c_stream_to_table_and_reader(fail_collect):
@@ -1591,16 +1593,16 @@ def test_arrow_c_stream_to_table_and_reader(fail_collect):
     batches = table.to_batches()
 
     assert len(batches) == 2
-    assert batches[0].equals(batch1)
-    assert batches[1].equals(batch2)
+    assert any(b.equals(batch1) for b in batches)
+    assert any(b.equals(batch2) for b in batches)
     assert table.schema == df.schema()
     assert table.column("a").num_chunks == 2
+    assert sorted(table.column("a").to_pylist()) == [1, 2]
 
     reader = pa.RecordBatchReader._import_from_c_capsule(df.__arrow_c_stream__())
     assert isinstance(reader, pa.RecordBatchReader)
     reader_table = pa.Table.from_batches(reader)
-    expected = pa.Table.from_batches([batch1, batch2])
-    assert reader_table.equals(expected)
+    assert sorted(reader_table.column("a").to_pylist()) == [1, 2]
 
 
 def test_arrow_c_stream_capsule_ownership(fail_collect):