more test

Author: dcherian

cf_xarray/tests/test_accessor.py

@@ -987,6 +987,7 @@ def test_get_bounds_dim_name() -> None:
         ds.cf.get_bounds_dim_name("longitude")
 
 
+@requires_pyproj
 def test_grid_mappings():
     ds = rotds.copy(deep=False)
 
@@ -1036,6 +1037,26 @@ def test_grid_mappings():
     # grid_mapping_name
     assert ds.cf["temp"].cf.grid_mapping_name == "rotated_latitude_longitude"
 
+    # Test .grid_mappings property with single grid mapping
+    grid_mappings = ds.cf.grid_mappings
+    assert len(grid_mappings) == 1
+    gm = grid_mappings[0]
+    assert gm.name == "rotated_latitude_longitude"
+    assert gm.array.name == "rotated_pole"
+    assert gm.array.shape == ()  # scalar variable
+    assert isinstance(gm.coordinates, tuple)
+    # Should have rlon and rlat detected from standard names
+    assert gm.coordinates == ("rlon", "rlat")
+
+    # Test .grid_mappings property on DataArray with propagated coords
+    da_grid_mappings = ds.cf["temp"].cf.grid_mappings
+    assert len(da_grid_mappings) == 1
+    da_gm = da_grid_mappings[0]
+    assert da_gm.name == "rotated_latitude_longitude"
+    assert da_gm.array.name == "rotated_pole"
+    # Should also detect rlon and rlat from standard names
+    assert da_gm.coordinates == ("rlon", "rlat")
+
     # what if there are really 2 grid mappins?
     ds["temp2"] = ds.temp
     ds["temp2"].attrs["grid_mapping"] = "rotated_pole2"
@@ -1240,6 +1261,7 @@ def test_grid_mappings_coordinates_attribute():
             )
 
 
+@requires_pyproj
 def test_bad_grid_mapping_attribute():
     ds = rotds.copy(deep=False)
     ds.temp.attrs["grid_mapping"] = "foo"
@@ -1255,6 +1277,10 @@ def test_bad_grid_mapping_attribute():
     with pytest.warns(UserWarning):
         ds.cf.get_associated_variable_names("temp", error=False)
 
+    # Test .grid_mappings property with bad grid mapping - should return empty tuple
+    grid_mappings = ds.cf.grid_mappings
+    assert grid_mappings == ()  # No valid grid mappings since 'foo' doesn't exist
+
 
 def test_docstring() -> None:
     assert "One of ('X'" in airds.cf.groupby.__doc__