test: add unit test for creating a Table from RecordBatch and validate schema

Author: kosiew

python/tests/test_io.py

@@ -135,3 +135,17 @@ def test_table_from_batches_stream(ctx, fail_collect):
 
     for batch in df:
         assert isinstance(batch, pa.RecordBatch)
+import pyarrow as pa
+
+
+def test_table_from_batches_stream():
+    schema = pa.schema([pa.field("value", pa.int64(), nullable=False)])
+    batch = pa.RecordBatch.from_arrays([pa.array(range(10))], schema=schema)
+    reader = pa.RecordBatchReader.from_batches(schema, [batch])
+    result = pa.Table.from_batches(reader)
+
+    expected_schema = pa.schema([pa.field("value", pa.int64(), nullable=False)])
+    expected = pa.Table.from_arrays([pa.array(range(10))], schema=expected_schema)
+
+    assert result == expected
+    assert result.schema == expected_schema