ogr_parquet.py: test behaviour on empty file (refs #8509)

OSGeo · Oct 3, 2023 · 3d0ba02 · 3d0ba02
1 parent 48f46f8
commit 3d0ba02
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/autotest/ogr/ogr_parquet.py b/autotest/ogr/ogr_parquet.py
@@ -1686,6 +1686,26 @@ def test_ogr_parquet_arrow_stream_numpy():
     )
 
 
+###############################################################################
+# Test reading an empty file with GetArrowStream()
+
+
+def test_ogr_parquet_arrow_stream_empty_file():
+
+    ds = ogr.GetDriverByName("Parquet").CreateDataSource("/vsimem/test.parquet")
+    ds.CreateLayer("test", geom_type=ogr.wkbPoint)
+    ds = None
+    ds = ogr.Open("/vsimem/test.parquet")
+    lyr = ds.GetLayer(0)
+    assert lyr.TestCapability(ogr.OLCFastGetArrowStream) == 1
+    stream = lyr.GetArrowStream()
+    assert stream.GetNextRecordBatch() is None
+    del stream
+    ds = None
+
+    ogr.GetDriverByName("Parquet").DeleteDataSource("/vsimem/test.parquet")
+
+
 ###############################################################################