Parameterize expected values

Author: redcliff

tests/test_xstac.py

@@ -51,22 +51,11 @@ def test_xarray_to_stac(
     assert dimensions == collection_expected_dims
     assert result.extra_fields["cube:variables"] == collection_expected_vars
 
-@pytest.mark.parametrize("dimension_key", ["x_dimension", "y_dimension", "temporal_dimension", "longitude", "latitude"])
-def test_maybe_use_cf_standard_axis(ds, dimension_key):
-    match dimension_key:
-        case "x_dimension":
-            expected = "x"
-        case "y_dimension":
-            expected = "y"
-        case "temporal_dimension":
-            expected = "time"
-        case "longitude":
-            expected = "lon"
-        case "latitude":
-            expected = "lat"
-        case _:
-            expected = dimension_key
-    assert maybe_use_cf_standard_axis(None, dimension_key, ds) == expected
+@pytest.mark.parametrize(
+    ("dimension_key", "expected_value"), 
+    [("x_dimension", "x"), ("y_dimension", "y"), ("temporal_dimension", "time"), ("longitude", "lon"), ("latitude", "lat")])
+def test_maybe_use_cf_standard_axis(ds, dimension_key, expected_value):
+    assert maybe_use_cf_standard_axis(None, dimension_key, ds) == expected_value
 
 def test_validation_with_none(ds_without_spatial_dims):
     # https://github.com/TomAugspurger/xstac/issues/9