tests: Add tests for normalization

tests/normalization_test.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+import pytest
+import xarray as xr
+
+from . import DATA_DIR
+from thalassa import api
+from thalassa import normalization
+from thalassa.normalization import THALASSA_FORMATS
+
+@pytest.mark.parametrize(
+    "ds,expected_fmt",
+    [
+        pytest.param(api.open_dataset(DATA_DIR / "fort.63.nc", normalize=False), THALASSA_FORMATS.ADCIRC, id="ADCIRC"),
+        pytest.param(xr.Dataset(), THALASSA_FORMATS.UNKNOWN, id="Unknown"),
+    ],
+)
+def test_infer_format(ds, expected_fmt):
+    fmt = normalization.infer_format(ds)
+    assert fmt == expected_fmt
+
+
+@pytest.mark.parametrize(
+    "path,expected",
+    [
+        pytest.param(DATA_DIR / "fort.63.nc", True, id="ADCIRC"),
+        pytest.param(__file__, False, id="Unknown"),
+    ],
+)
+def test_can_be_inferred(path, expected):
+    result = normalization.can_be_inferred(path)
+    assert result == expected

thalassa/normalization.py

@@ -87,12 +87,12 @@ def is_adcirc(ds: xarray.Dataset) -> bool:
 def infer_format(ds: xarray.Dataset) -> THALASSA_FORMATS:
     if is_schism(ds):
         fmt = THALASSA_FORMATS.SCHISM
-    elif is_adcirc(ds):
-        fmt = THALASSA_FORMATS.ADCIRC
     elif is_pyposeidon(ds):
         fmt = THALASSA_FORMATS.PYPOSEIDON
     elif is_generic(ds):
         fmt = THALASSA_FORMATS.GENERIC
+    elif is_adcirc(ds):
+        fmt = THALASSA_FORMATS.ADCIRC
     else:
         fmt = THALASSA_FORMATS.UNKNOWN
     logger.debug("Inferred format: %s", fmt)