roms-tools 1.6.1__py3-none-any.whl → 1.7.0__py3-none-any.whl

roms_tools/tests/test_setup/test_data/river_forcing.zarr/river_volume/.zattrs

@@ -0,0 +1,9 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "river_time",
+        "nriver"
+    ],
+    "coordinates": "abs_time month river_name",
+    "long_name": "River volume flux",
+    "units": "m^3/s"
+}

roms_tools/tests/test_setup/test_data/river_forcing.zarr/tracer_name/.zarray

@@ -0,0 +1,20 @@
+{
+    "chunks": [
+        2
+    ],
+    "compressor": {
+        "blocksize": 0,
+        "clevel": 5,
+        "cname": "lz4",
+        "id": "blosc",
+        "shuffle": 1
+    },
+    "dtype": "<U11",
+    "fill_value": null,
+    "filters": null,
+    "order": "C",
+    "shape": [
+        2
+    ],
+    "zarr_format": 2
+}

roms_tools/tests/test_setup/test_data/river_forcing.zarr/tracer_name/.zattrs

@@ -0,0 +1,6 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "ntracers"
+    ],
+    "long_name": "Tracer name"
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/.zattrs

@@ -0,0 +1 @@
+{}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/.zgroup

@@ -0,0 +1,3 @@
+{
+    "zarr_format": 2
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/.zmetadata

@@ -0,0 +1,185 @@
+{
+    "metadata": {
+        ".zattrs": {},
+        ".zgroup": {
+            "zarr_format": 2
+        },
+        "abs_time/.zarray": {
+            "chunks": [
+                4
+            ],
+            "compressor": {
+                "blocksize": 0,
+                "clevel": 5,
+                "cname": "lz4",
+                "id": "blosc",
+                "shuffle": 1
+            },
+            "dtype": "<i8",
+            "fill_value": null,
+            "filters": null,
+            "order": "C",
+            "shape": [
+                4
+            ],
+            "zarr_format": 2
+        },
+        "abs_time/.zattrs": {
+            "_ARRAY_DIMENSIONS": [
+                "river_time"
+            ],
+            "calendar": "proleptic_gregorian",
+            "long_name": "absolute time",
+            "units": "days since 1997-12-15 00:00:00"
+        },
+        "river_name/.zarray": {
+            "chunks": [
+                6
+            ],
+            "compressor": {
+                "blocksize": 0,
+                "clevel": 5,
+                "cname": "lz4",
+                "id": "blosc",
+                "shuffle": 1
+            },
+            "dtype": "|O",
+            "fill_value": null,
+            "filters": [
+                {
+                    "id": "vlen-utf8"
+                }
+            ],
+            "order": "C",
+            "shape": [
+                6
+            ],
+            "zarr_format": 2
+        },
+        "river_name/.zattrs": {
+            "_ARRAY_DIMENSIONS": [
+                "nriver"
+            ],
+            "long_name": "River name"
+        },
+        "river_time/.zarray": {
+            "chunks": [
+                4
+            ],
+            "compressor": {
+                "blocksize": 0,
+                "clevel": 5,
+                "cname": "lz4",
+                "id": "blosc",
+                "shuffle": 1
+            },
+            "dtype": "<f8",
+            "fill_value": "NaN",
+            "filters": null,
+            "order": "C",
+            "shape": [
+                4
+            ],
+            "zarr_format": 2
+        },
+        "river_time/.zattrs": {
+            "_ARRAY_DIMENSIONS": [
+                "river_time"
+            ],
+            "long_name": "relative time: days since 2000-01-01 00:00:00",
+            "units": "days"
+        },
+        "river_tracer/.zarray": {
+            "chunks": [
+                4,
+                2,
+                6
+            ],
+            "compressor": {
+                "blocksize": 0,
+                "clevel": 5,
+                "cname": "lz4",
+                "id": "blosc",
+                "shuffle": 1
+            },
+            "dtype": "<f4",
+            "fill_value": "NaN",
+            "filters": null,
+            "order": "C",
+            "shape": [
+                4,
+                2,
+                6
+            ],
+            "zarr_format": 2
+        },
+        "river_tracer/.zattrs": {
+            "_ARRAY_DIMENSIONS": [
+                "river_time",
+                "ntracers",
+                "nriver"
+            ],
+            "coordinates": "abs_time river_name tracer_name",
+            "long_name": "River tracer data",
+            "units": "degrees C [temperature]; psu [salinity]"
+        },
+        "river_volume/.zarray": {
+            "chunks": [
+                4,
+                6
+            ],
+            "compressor": {
+                "blocksize": 0,
+                "clevel": 5,
+                "cname": "lz4",
+                "id": "blosc",
+                "shuffle": 1
+            },
+            "dtype": "<f4",
+            "fill_value": "NaN",
+            "filters": null,
+            "order": "C",
+            "shape": [
+                4,
+                6
+            ],
+            "zarr_format": 2
+        },
+        "river_volume/.zattrs": {
+            "_ARRAY_DIMENSIONS": [
+                "river_time",
+                "nriver"
+            ],
+            "coordinates": "abs_time river_name",
+            "long_name": "River volume flux",
+            "units": "m^3/s"
+        },
+        "tracer_name/.zarray": {
+            "chunks": [
+                2
+            ],
+            "compressor": {
+                "blocksize": 0,
+                "clevel": 5,
+                "cname": "lz4",
+                "id": "blosc",
+                "shuffle": 1
+            },
+            "dtype": "<U11",
+            "fill_value": null,
+            "filters": null,
+            "order": "C",
+            "shape": [
+                2
+            ],
+            "zarr_format": 2
+        },
+        "tracer_name/.zattrs": {
+            "_ARRAY_DIMENSIONS": [
+                "ntracers"
+            ],
+            "long_name": "Tracer name"
+        }
+    },
+    "zarr_consolidated_format": 1
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/abs_time/.zarray

@@ -0,0 +1,20 @@
+{
+    "chunks": [
+        4
+    ],
+    "compressor": {
+        "blocksize": 0,
+        "clevel": 5,
+        "cname": "lz4",
+        "id": "blosc",
+        "shuffle": 1
+    },
+    "dtype": "<i8",
+    "fill_value": null,
+    "filters": null,
+    "order": "C",
+    "shape": [
+        4
+    ],
+    "zarr_format": 2
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/abs_time/.zattrs

@@ -0,0 +1,8 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "river_time"
+    ],
+    "calendar": "proleptic_gregorian",
+    "long_name": "absolute time",
+    "units": "days since 1997-12-15 00:00:00"
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/river_name/.zarray

@@ -0,0 +1,24 @@
+{
+    "chunks": [
+        6
+    ],
+    "compressor": {
+        "blocksize": 0,
+        "clevel": 5,
+        "cname": "lz4",
+        "id": "blosc",
+        "shuffle": 1
+    },
+    "dtype": "|O",
+    "fill_value": null,
+    "filters": [
+        {
+            "id": "vlen-utf8"
+        }
+    ],
+    "order": "C",
+    "shape": [
+        6
+    ],
+    "zarr_format": 2
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/river_name/.zattrs

@@ -0,0 +1,6 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "nriver"
+    ],
+    "long_name": "River name"
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/river_time/.zarray

@@ -0,0 +1,20 @@
+{
+    "chunks": [
+        4
+    ],
+    "compressor": {
+        "blocksize": 0,
+        "clevel": 5,
+        "cname": "lz4",
+        "id": "blosc",
+        "shuffle": 1
+    },
+    "dtype": "<f8",
+    "fill_value": "NaN",
+    "filters": null,
+    "order": "C",
+    "shape": [
+        4
+    ],
+    "zarr_format": 2
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/river_time/.zattrs

@@ -0,0 +1,7 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "river_time"
+    ],
+    "long_name": "relative time: days since 2000-01-01 00:00:00",
+    "units": "days"
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/river_tracer/.zarray

@@ -0,0 +1,24 @@
+{
+    "chunks": [
+        4,
+        2,
+        6
+    ],
+    "compressor": {
+        "blocksize": 0,
+        "clevel": 5,
+        "cname": "lz4",
+        "id": "blosc",
+        "shuffle": 1
+    },
+    "dtype": "<f4",
+    "fill_value": "NaN",
+    "filters": null,
+    "order": "C",
+    "shape": [
+        4,
+        2,
+        6
+    ],
+    "zarr_format": 2
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/river_tracer/.zattrs

@@ -0,0 +1,10 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "river_time",
+        "ntracers",
+        "nriver"
+    ],
+    "coordinates": "abs_time river_name tracer_name",
+    "long_name": "River tracer data",
+    "units": "degrees C [temperature]; psu [salinity]"
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/river_volume/.zarray

@@ -0,0 +1,22 @@
+{
+    "chunks": [
+        4,
+        6
+    ],
+    "compressor": {
+        "blocksize": 0,
+        "clevel": 5,
+        "cname": "lz4",
+        "id": "blosc",
+        "shuffle": 1
+    },
+    "dtype": "<f4",
+    "fill_value": "NaN",
+    "filters": null,
+    "order": "C",
+    "shape": [
+        4,
+        6
+    ],
+    "zarr_format": 2
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/river_volume/.zattrs

@@ -0,0 +1,9 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "river_time",
+        "nriver"
+    ],
+    "coordinates": "abs_time river_name",
+    "long_name": "River volume flux",
+    "units": "m^3/s"
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/tracer_name/.zarray

@@ -0,0 +1,20 @@
+{
+    "chunks": [
+        2
+    ],
+    "compressor": {
+        "blocksize": 0,
+        "clevel": 5,
+        "cname": "lz4",
+        "id": "blosc",
+        "shuffle": 1
+    },
+    "dtype": "<U11",
+    "fill_value": null,
+    "filters": null,
+    "order": "C",
+    "shape": [
+        2
+    ],
+    "zarr_format": 2
+}

roms_tools/tests/test_setup/test_data/river_forcing_no_climatology.zarr/tracer_name/.zattrs

@@ -0,0 +1,6 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "ntracers"
+    ],
+    "long_name": "Tracer name"
+}

roms_tools/tests/test_setup/test_datasets.py

@@ -1,4 +1,5 @@
 import pytest
+import logging
 from datetime import datetime
 import numpy as np
 import xarray as xr
@@ -229,12 +230,12 @@ def test_multiple_matching_times(
     assert dataset.ds["time"].values == np.datetime64(datetime(2022, 2, 1, 0, 0))
 
 
-def test_warnings_times(global_dataset, tmp_path, use_dask):
+def test_warnings_times(global_dataset, tmp_path, caplog, use_dask):
     """Test handling when no matching times are found."""
     # Create a temporary file
     filepath = tmp_path / "test.nc"
     global_dataset.to_netcdf(filepath)
-    with pytest.warns(Warning, match="No records found at or before the start_time."):
+    with caplog.at_level(logging.WARNING):
         start_time = datetime(2021, 1, 1)
         end_time = datetime(2021, 2, 1)
 
@@ -245,8 +246,10 @@ def test_warnings_times(global_dataset, tmp_path, use_dask):
             end_time=end_time,
             use_dask=use_dask,
         )
+    # Verify the warning message in the log
+    assert "No records found at or before the start_time." in caplog.text
 
-    with pytest.warns(Warning, match="No records found at or after the end_time."):
+    with caplog.at_level(logging.WARNING):
         start_time = datetime(2024, 1, 1)
         end_time = datetime(2024, 2, 1)
 
@@ -257,6 +260,8 @@ def test_warnings_times(global_dataset, tmp_path, use_dask):
             end_time=end_time,
             use_dask=use_dask,
         )
+    # Verify the warning message in the log
+    assert "No records found at or after the end_time." in caplog.text
 
 
 def test_from_ds(global_dataset, global_dataset_with_noon_times, use_dask, tmp_path):

roms_tools/tests/test_setup/test_grid.py

@@ -1,4 +1,5 @@
 import pytest
+import logging
 import xarray as xr
 from roms_tools import Grid
 import importlib.metadata
@@ -390,7 +391,7 @@ def test_from_yaml_missing_grid(tmp_path):
         yaml_filepath.unlink()
 
 
-def test_from_yaml_version_mismatch(tmp_path):
+def test_from_yaml_version_mismatch(tmp_path, caplog):
     yaml_content = textwrap.dedent(
         """\
     ---
@@ -423,11 +424,11 @@ def test_from_yaml_version_mismatch(tmp_path):
             with open(yaml_filepath, "w") as f:
                 f.write(yaml_content)
 
-        with pytest.warns(
-            UserWarning,
-            match="Current roms-tools version.*does not match the version in the YAML header.*",
-        ):
+        with caplog.at_level(logging.WARNING):
             Grid.from_yaml(yaml_filepath)
 
+        # Verify the warning message in the log
+        assert "Current roms-tools version" in caplog.text
+
         yaml_filepath = Path(yaml_filepath)
         yaml_filepath.unlink()

roms_tools/tests/test_setup/test_initial_conditions.py

@@ -309,9 +309,7 @@ def test_from_yaml_missing_initial_conditions(tmp_path, use_dask):
       center_lat: 61
       rot: -20
       topography_source: ETOPO5
-      smooth_factor: 8
       hmin: 5.0
-      rmax: 0.2
     """
     )