fix datetime plotting (#125)

Author: martinfleis

xvec/plotting.py

@@ -5,6 +5,7 @@
 import shapely
 import xarray as xr
 import xproj  # noqa: F401
+from xarray.core.formatting import format_item
 from xarray.plot.utils import _determine_cmap_params, label_from_attrs
 
 
@@ -122,7 +123,7 @@ def _plot_faceted(arr, axs, row, col, hue, geometry, cmap_params=None, **kwargs)
         for i_r, row_val in enumerate(arr[row]):
             for i_c, col_val in enumerate(arr[col]):
                 _plot_single_panel(
-                    arr.sel({row: [row_val.item()], col: [col_val.item()]}).drop_vars(
+                    arr.sel({row: [row_val.values], col: [col_val.values]}).drop_vars(
                         [col, row]
                     ),
                     axs[i_r, i_c],
@@ -133,12 +134,13 @@ def _plot_faceted(arr, axs, row, col, hue, geometry, cmap_params=None, **kwargs)
                 )
                 if i_r == 0:
                     axs[0, i_c].set_title(
-                        f"{col} = {arr[col][i_c].astype(str).item()}", fontsize="small"
+                        f"{col} = {format_item(arr[col][i_c].values)}",
+                        fontsize="small",
                     )
                 if i_c == len(arr[col]) - 1:
                     axs[i_r, -1].yaxis.set_label_position("right")
                     axs[i_r, -1].set_ylabel(
-                        f"{row} = {arr[row][i_r].item()}",
+                        f"{row} = {format_item(arr[row][i_r].values)}",
                         fontsize="small",
                         rotation=270,
                         labelpad=12,
@@ -155,7 +157,7 @@ def _plot_faceted(arr, axs, row, col, hue, geometry, cmap_params=None, **kwargs)
                 **kwargs,
             )
             axs_flat[i_c].set_title(
-                f"{col} = {arr[col][i_c].astype(str).item()}", fontsize="small"
+                f"{col} = {format_item(arr[col][i_c].values)}", fontsize="small"
             )
         return arr[col].shape[0]  # Return used axes count
 

xvec/tests/baseline_images/test_plotting/categorical.png

@@ -1,5 +1,6 @@
 import geodatasets
 import geopandas as gpd
+import pandas as pd
 import pytest
 import xarray as xr
 import xproj  # noqa: F401
@@ -30,6 +31,7 @@ def glaciers():
     glaciers_df = gpd.read_file(
         "https://github.com/loreabad6/post/raw/refs/heads/main/inst/extdata/svalbard.gpkg"
     )
+    glaciers_df["year"] = pd.to_datetime(glaciers_df["year"].astype(int), format="%Y")
     return (
         glaciers_df.set_index(["name", "year"])
         .to_xarray()
@@ -117,7 +119,7 @@ def test_var_geom(glaciers):
     ax0 = ax[0][0]
     assert ax0.get_xlabel() == "Easting\n[metre]"
     assert ax0.get_ylabel() == "Northing\n[metre]"
-    assert ax0.get_title() == "year = 1936.0"
+    assert ax0.get_title() == "year = 1936-01-01"
 
 
 @image_comparison(
@@ -130,7 +132,7 @@ def test_var_geom_facet(glaciers):
     assert ax[2][0].get_xlabel() == "Easting\n[metre]"
     assert ax[0][0].get_ylabel() == "Northing\n[metre]"
     assert ax[0][0].get_title() == "name = Austre Brøggerbreen"
-    assert ax[0][-1].get_ylabel() == "year = 1936.0"
+    assert ax[0][-1].get_ylabel() == "year = 1936-01-01"
 
 
 @image_comparison(
@@ -143,7 +145,7 @@ def test_var_geom_ds(glaciers):
     ax0 = ax[0][0]
     assert ax0.get_xlabel() == "Easting\n[metre]"
     assert ax0.get_ylabel() == "Northing\n[metre]"
-    assert ax0.get_title() == "year = 1936.0"
+    assert ax0.get_title() == "year = 1936-01-01"
 
 
 @image_comparison(baseline_images=["hue"], extensions=["png"], style=[], tol=0.01)
@@ -154,7 +156,7 @@ def test_hue(glaciers):
     ax0 = ax[0][0]
     assert ax0.get_xlabel() == "Easting\n[metre]"
     assert ax0.get_ylabel() == "Northing\n[metre]"
-    assert ax0.get_title() == "year = 1936.0"
+    assert ax0.get_title() == "year = 1936-01-01"
 
 
 @image_comparison(
@@ -183,7 +185,7 @@ def test_categorical(glaciers):
     ax0 = ax[0][0]
     assert ax0.get_xlabel() == "Easting\n[metre]"
     assert ax0.get_ylabel() == "Northing\n[metre]"
-    assert ax0.get_title() == "year = 1936.0"
+    assert ax0.get_title() == "year = 1936-01-01"
 
 
 @image_comparison(