Add unit test for plotting GeoDataFrame with int32 and int64 columns

Using the @RidgeTest.shp dataset that has been buffered, and colouring the polygons with a different colormap.

Author: weiji14

pygmt/tests/baseline/test_geopandas_plot_int_dtypes.png.dvc

@@ -0,0 +1,4 @@
+outs:
+- md5: 6ce1b73d25ac10900a2ce4c7148d2953
+  size: 43434
+  path: test_geopandas_plot_int_dtypes.png

pygmt/tests/test_geopandas.py

@@ -2,8 +2,9 @@
 Test integration with geopandas.
 """
 import numpy.testing as npt
+import pandas as pd
 import pytest
-from pygmt import Figure, info
+from pygmt import Figure, info, which, makecpt
 
 gpd = pytest.importorskip("geopandas")
 shapely = pytest.importorskip("shapely")
@@ -131,3 +132,44 @@ def test_geopandas_plot3d_non_default_circle():
         style="c0.2c",
     )
     return fig
+
+
+@pytest.mark.parametrize("dtype", ["int32", "int64", pd.Int32Dtype(), pd.Int64Dtype()])
+@pytest.mark.mpl_image_compare(filename="test_geopandas_plot_int_dtypes.png")
+def test_geopandas_plot_int_dtypes(dtype):
+    """
+    Check that plotting a geopandas GeoDataFrame with integer columns works,
+    including int32 and int64 (non-nullable), Int32 and Int64 (nullable).
+
+    This is a regression test for
+    https://github.com/GenericMappingTools/pygmt/issues/2497
+    """
+    # Read shapefile in geopandas.GeoDataFrame
+    shapefile = which(
+        fname="@RidgeTest.shp @RidgeTest.shx @RidgeTest.dbf @RidgeTest.prj",
+        download="c",
+    )
+    gdf = gpd.read_file(shapefile[0])
+
+    # Reproject geometry and change dtype of NPOINTS column
+    gdf["geometry"] = (
+        gdf.to_crs(crs="EPSG:3857")
+        .buffer(distance=100000)
+        .to_crs(crs="OGC:CRS84")  # convert to lon/lat to prevent @null in PROJ CRS
+    )
+    gdf["NPOINTS"] = gdf.NPOINTS.astype(dtype=dtype)
+
+    # Plot figure with three polygons colored based on NPOINTS value
+    fig = Figure()
+    makecpt(cmap="lajolla", series=[10, 60, 10], continuous=True)
+    fig.plot(
+        data=gdf,
+        frame=True,
+        pen="1p,black",
+        close=True,
+        fill="+z",
+        cmap=True,
+        aspatial="Z=NPOINTS",
+    )
+    fig.colorbar()
+    return fig