mosaic reads geoparquet

databrickslabs · Nov 28, 2024 · 5306ad7 · 5306ad7
1 parent 5a25f8e
commit 5306ad7
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/python/test/data/example.parquet b/python/test/data/example.parquet
diff --git a/python/test/test_raster_functions.py b/python/test/test_raster_functions.py
@@ -321,3 +321,16 @@ def test_dtmfromgeoms(self):
         )
 
         result.write.mode("overwrite").format("noop").save()
+
+
+    def test_geoparquet_reader(self):
+        geoparquet_df = (
+            readers.read()
+            .format("multi_read_ogr")
+            .option("driverName", "parquet")
+            .option("chunkSize", "2")
+            .load("test/data/example.parquet")
+        )
+
+        self.assertEqual(geoparquet_df.count(), 5)
+