Add test for ordering of record batches in arrow C stream

Author: kosiew

python/tests/test_dataframe.py

@@ -1610,6 +1610,23 @@ def test_arrow_c_stream_to_table(fail_collect):
     assert table.column("a").num_chunks == 2
 
 
+def test_arrow_c_stream_order():
+    ctx = SessionContext()
+
+    batch1 = pa.record_batch([pa.array([1])], names=["a"])
+    batch2 = pa.record_batch([pa.array([2])], names=["a"])
+
+    df = ctx.create_dataframe([[batch1, batch2]])
+
+    table = pa.Table.from_batches(df)
+    expected = pa.Table.from_batches([batch1, batch2])
+
+    assert table.equals(expected)
+    col = table.column("a")
+    assert col.chunk(0)[0].as_py() == 1
+    assert col.chunk(1)[0].as_py() == 2
+
+
 def test_arrow_c_stream_reader(df):
     reader = pa.RecordBatchReader._import_from_c_capsule(df.__arrow_c_stream__())
     assert isinstance(reader, pa.RecordBatchReader)