Fix: kwargs not passed on to OGR in write_dataframe (#67)

CHANGES.md

@@ -27,6 +27,7 @@
 
 -   use dtype `object` instead of `numpy.object` to eliminate deprecation warnings (#34)
 -   raise error if layer cannot be opened (#35)
+-   fix passing gdal creation parameters in `write_dataframe` (#62)
 
 ## 0.3.0
 

pyogrio/geopandas.py

@@ -216,4 +216,5 @@ def write_dataframe(df, path, layer=None, driver=None, encoding=None, **kwargs):
         crs=crs,
         geometry_type=geometry_type,
         encoding=encoding,
+        **kwargs
     )

pyogrio/tests/test_geopandas_io.py

@@ -238,6 +238,22 @@ def test_write_empty_dataframe(tmpdir, driver, ext):
     assert_geodataframe_equal(df, expected)
 
 
+def test_write_dataframe_gdalparams(tmpdir, naturalearth_lowres):
+    original_df = read_dataframe(naturalearth_lowres)
+
+    test_noindex_filename = os.path.join(str(tmpdir), f"test_gdalparams_noindex.shp")
+    write_dataframe(original_df, test_noindex_filename, SPATIAL_INDEX="NO")
+    assert os.path.exists(test_noindex_filename) is True
+    test_noindex_index_filename = os.path.join(str(tmpdir), f"test_gdalparams_noindex.qix")
+    assert os.path.exists(test_noindex_index_filename) is False
+
+    test_withindex_filename = os.path.join(str(tmpdir), f"test_gdalparams_withindex.shp")
+    write_dataframe(original_df, test_withindex_filename, SPATIAL_INDEX="YES")
+    assert os.path.exists(test_withindex_filename) is True
+    test_withindex_index_filename = os.path.join(str(tmpdir), f"test_gdalparams_withindex.qix")
+    assert os.path.exists(test_withindex_index_filename) is True
+
+
 @pytest.mark.filterwarnings(
     "ignore: You will likely lose important projection information"
 )