We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f9b23d9 commit 531cd36Copy full SHA for 531cd36
1 file changed
python/tests/test_dataframe.py
@@ -1648,8 +1648,9 @@ def test_arrow_c_stream_to_table_and_reader(fail_collect):
1648
batches = table.to_batches()
1649
1650
assert len(batches) == 2
1651
- assert batches[0].equals(batch1)
1652
- assert batches[1].equals(batch2)
+ expected = [batch1, batch2]
+ for exp in expected:
1653
+ assert any(got.equals(exp) for got in batches)
1654
assert table.schema == df.schema()
1655
assert table.column("a").num_chunks == 2
1656
0 commit comments