Skip to content

Commit 2e4b963

Browse files
committed
test: enhance test_arrow_c_stream_to_table_and_reader for batch equality validation
1 parent 1ebd3c1 commit 2e4b963

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

python/tests/test_dataframe.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1648,8 +1648,9 @@ def test_arrow_c_stream_to_table_and_reader(fail_collect):
16481648
batches = table.to_batches()
16491649

16501650
assert len(batches) == 2
1651-
assert batches[0].equals(batch1)
1652-
assert batches[1].equals(batch2)
1651+
expected = [batch1, batch2]
1652+
for exp in expected:
1653+
assert any(got.equals(exp) for got in batches)
16531654
assert table.schema == df.schema()
16541655
assert table.column("a").num_chunks == 2
16551656

0 commit comments

Comments
 (0)