test: add unit test for converting DataFrame to Arrow Table from record batches

Author: kosiew

python/tests/test_arrow_interop.py

@@ -0,0 +1,16 @@
+import pyarrow as pa
+import pytest
+
+
+def test_table_from_batches_with_dataframe(ctx):
+    batch1 = pa.record_batch({"a": pa.array([1, 2]), "b": pa.array(["x", "y"])})
+    batch2 = pa.record_batch({"a": pa.array([3, 4]), "b": pa.array(["z", "w"])})
+    df = ctx.create_dataframe([[batch1], [batch2]])
+
+    try:
+        table = pa.Table.from_batches(df)
+    except TypeError as err:  # pragma: no cover - failure path
+        pytest.fail(f"TypeError raised when converting DataFrame to Arrow Table: {err}")
+
+    expected = pa.Table.from_batches([batch1, batch2])
+    assert table.equals(expected)