roms-tools 2.2.1__py3-none-any.whl → 2.4.0__py3-none-any.whl

roms_tools/tests/test_setup/test_data/boundary_forcing.zarr/zeta_east/.zarray

@@ -1,7 +1,7 @@
 {
     "chunks": [
-        1,
-        4
+        2,
+        5
     ],
     "compressor": {
         "blocksize": 0,
@@ -15,8 +15,8 @@
     "filters": null,
     "order": "C",
     "shape": [
-        1,
-        4
+        2,
+        5
     ],
     "zarr_format": 2
 }

roms_tools/tests/test_setup/test_data/boundary_forcing.zarr/zeta_east/.zattrs

@@ -3,7 +3,15 @@
         "bry_time",
         "eta_rho"
     ],
+    "_ChunkSizes": [
+        1,
+        681,
+        1440
+    ],
+    "cell_methods": "area: mean",
     "coordinates": "abs_time",
     "long_name": "eastern boundary sea surface height",
+    "standard_name": "sea_surface_height_above_geoid",
+    "unit_long": "Meters",
     "units": "m"
 }

roms_tools/tests/test_setup/test_data/boundary_forcing.zarr/zeta_north/.zarray

@@ -1,7 +1,7 @@
 {
     "chunks": [
-        1,
-        4
+        2,
+        5
     ],
     "compressor": {
         "blocksize": 0,
@@ -15,8 +15,8 @@
     "filters": null,
     "order": "C",
     "shape": [
-        1,
-        4
+        2,
+        5
     ],
     "zarr_format": 2
 }

roms_tools/tests/test_setup/test_data/boundary_forcing.zarr/zeta_north/.zattrs

@@ -3,7 +3,15 @@
         "bry_time",
         "xi_rho"
     ],
+    "_ChunkSizes": [
+        1,
+        681,
+        1440
+    ],
+    "cell_methods": "area: mean",
     "coordinates": "abs_time",
     "long_name": "northern boundary sea surface height",
+    "standard_name": "sea_surface_height_above_geoid",
+    "unit_long": "Meters",
     "units": "m"
 }

roms_tools/tests/test_setup/test_data/boundary_forcing.zarr/zeta_south/.zarray

@@ -1,7 +1,7 @@
 {
     "chunks": [
-        1,
-        4
+        2,
+        5
     ],
     "compressor": {
         "blocksize": 0,
@@ -15,8 +15,8 @@
     "filters": null,
     "order": "C",
     "shape": [
-        1,
-        4
+        2,
+        5
     ],
     "zarr_format": 2
 }

roms_tools/tests/test_setup/test_data/boundary_forcing.zarr/zeta_south/.zattrs

@@ -3,7 +3,15 @@
         "bry_time",
         "xi_rho"
     ],
+    "_ChunkSizes": [
+        1,
+        681,
+        1440
+    ],
+    "cell_methods": "area: mean",
     "coordinates": "abs_time",
     "long_name": "southern boundary sea surface height",
+    "standard_name": "sea_surface_height_above_geoid",
+    "unit_long": "Meters",
     "units": "m"
 }

roms_tools/tests/test_setup/test_data/boundary_forcing.zarr/zeta_west/.zarray

@@ -1,7 +1,7 @@
 {
     "chunks": [
-        1,
-        4
+        2,
+        5
     ],
     "compressor": {
         "blocksize": 0,
@@ -15,8 +15,8 @@
     "filters": null,
     "order": "C",
     "shape": [
-        1,
-        4
+        2,
+        5
     ],
     "zarr_format": 2
 }

roms_tools/tests/test_setup/test_data/boundary_forcing.zarr/zeta_west/.zattrs

@@ -3,7 +3,15 @@
         "bry_time",
         "eta_rho"
     ],
+    "_ChunkSizes": [
+        1,
+        681,
+        1440
+    ],
+    "cell_methods": "area: mean",
     "coordinates": "abs_time",
     "long_name": "western boundary sea surface height",
+    "standard_name": "sea_surface_height_above_geoid",
+    "unit_long": "Meters",
     "units": "m"
 }

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

@@ -1,12 +1,12 @@
 {
     "center_lat": 0,
-    "center_lon": 0,
+    "center_lon": 0.5,
     "hc": 300.0,
     "hmin": 5.0,
-    "roms_tools_version": "0.1.dev239+dirty",
+    "roms_tools_version": "0.1.dev180+gfb7e1ef.d20250106",
     "rot": 20,
-    "size_x": 100,
-    "size_y": 100,
+    "size_x": 1000,
+    "size_y": 1000,
     "theta_b": 2.0,
     "theta_s": 5.0,
     "title": "ROMS grid created by ROMS-Tools",

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

@@ -2,13 +2,13 @@
     "metadata": {
         ".zattrs": {
             "center_lat": 0,
-            "center_lon": 0,
+            "center_lon": 0.5,
             "hc": 300.0,
             "hmin": 5.0,
-            "roms_tools_version": "0.1.dev239+dirty",
+            "roms_tools_version": "0.1.dev180+gfb7e1ef.d20250106",
             "rot": 20,
-            "size_x": 100,
-            "size_y": 100,
+            "size_x": 1000,
+            "size_y": 1000,
             "theta_b": 2.0,
             "theta_s": 5.0,
             "title": "ROMS grid created by ROMS-Tools",

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

@@ -1,9 +1,10 @@
 {
+    "adjust_depth_for_sea_surface_height": "False",
     "bgc_source": "CESM_REGRIDDED",
     "hc": 250.0,
     "ini_time": "2021-06-29 00:00:00",
     "model_reference_date": "2000-01-01 00:00:00",
-    "roms_tools_version": "0.1.dev157+dirty",
+    "roms_tools_version": "0.1.dev177",
     "source": "GLORYS",
     "theta_b": 2.0,
     "theta_s": 5.0,

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

@@ -1,11 +1,12 @@
 {
     "metadata": {
         ".zattrs": {
+            "adjust_depth_for_sea_surface_height": "False",
             "bgc_source": "CESM_REGRIDDED",
             "hc": 250.0,
             "ini_time": "2021-06-29 00:00:00",
             "model_reference_date": "2000-01-01 00:00:00",
-            "roms_tools_version": "0.1.dev157+dirty",
+            "roms_tools_version": "0.1.dev177",
             "source": "GLORYS",
             "theta_b": 2.0,
             "theta_s": 5.0,
@@ -112,7 +113,7 @@
             "_ARRAY_DIMENSIONS": [
                 "s_rho"
             ],
-            "long_name": "S-coordinate stretching curves at rho-points",
+            "long_name": "Vertical stretching function at rho-points",
             "units": "nondimensional"
         },
         "Cs_w/.zarray": {
@@ -139,7 +140,7 @@
             "_ARRAY_DIMENSIONS": [
                 "s_w"
             ],
-            "long_name": "S-coordinate stretching curves at w-points",
+            "long_name": "Vertical stretching function at w-points",
             "units": "nondimensional"
         },
         "DIC/.zarray": {
@@ -722,6 +723,7 @@
                 "ocean_time"
             ],
             "calendar": "proleptic_gregorian",
+            "long_name": "absolute time",
             "units": "days since 2021-06-29 00:00:00"
         },
         "diatC/.zarray": {
@@ -1081,7 +1083,7 @@
             "_ARRAY_DIMENSIONS": [
                 "ocean_time"
             ],
-            "long_name": "seconds since 2000-01-01 00:00:00",
+            "long_name": "relative time: seconds since 2000-01-01 00:00:00",
             "units": "seconds"
         },
         "salt/.zarray": {

roms_tools/tests/test_setup/test_data/initial_conditions_with_bgc_from_climatology.zarr/Cs_r/.zattrs

@@ -2,6 +2,6 @@
     "_ARRAY_DIMENSIONS": [
         "s_rho"
     ],
-    "long_name": "S-coordinate stretching curves at rho-points",
+    "long_name": "Vertical stretching function at rho-points",
     "units": "nondimensional"
 }

roms_tools/tests/test_setup/test_data/initial_conditions_with_bgc_from_climatology.zarr/Cs_w/.zattrs

@@ -2,6 +2,6 @@
     "_ARRAY_DIMENSIONS": [
         "s_w"
     ],
-    "long_name": "S-coordinate stretching curves at w-points",
+    "long_name": "Vertical stretching function at w-points",
     "units": "nondimensional"
 }

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

@@ -3,5 +3,6 @@
         "ocean_time"
     ],
     "calendar": "proleptic_gregorian",
+    "long_name": "absolute time",
     "units": "days since 2021-06-29 00:00:00"
 }

roms_tools/tests/test_setup/test_data/initial_conditions_with_bgc_from_climatology.zarr/ocean_time/.zattrs

@@ -2,6 +2,6 @@
     "_ARRAY_DIMENSIONS": [
         "ocean_time"
     ],
-    "long_name": "seconds since 2000-01-01 00:00:00",
+    "long_name": "relative time: seconds since 2000-01-01 00:00:00",
     "units": "seconds"
 }

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

@@ -32,6 +32,36 @@
             "long_name": "absolute time",
             "units": "days since 1997-12-15 00:00:00"
         },
+        "river_location/.zarray": {
+            "chunks": [
+                20,
+                20
+            ],
+            "compressor": {
+                "blocksize": 0,
+                "clevel": 5,
+                "cname": "lz4",
+                "id": "blosc",
+                "shuffle": 1
+            },
+            "dtype": "<f8",
+            "fill_value": "NaN",
+            "filters": null,
+            "order": "C",
+            "shape": [
+                20,
+                20
+            ],
+            "zarr_format": 2
+        },
+        "river_location/.zattrs": {
+            "_ARRAY_DIMENSIONS": [
+                "eta_rho",
+                "xi_rho"
+            ],
+            "long_name": "River index plus local volume fraction",
+            "units": "none"
+        },
         "river_name/.zarray": {
             "chunks": [
                 6

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

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

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

@@ -0,0 +1,8 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "eta_rho",
+        "xi_rho"
+    ],
+    "long_name": "River index plus local volume fraction",
+    "units": "none"
+}

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

@@ -60,6 +60,36 @@
             ],
             "long_name": "Month index (1-12)"
         },
+        "river_location/.zarray": {
+            "chunks": [
+                20,
+                20
+            ],
+            "compressor": {
+                "blocksize": 0,
+                "clevel": 5,
+                "cname": "lz4",
+                "id": "blosc",
+                "shuffle": 1
+            },
+            "dtype": "<f8",
+            "fill_value": "NaN",
+            "filters": null,
+            "order": "C",
+            "shape": [
+                20,
+                20
+            ],
+            "zarr_format": 2
+        },
+        "river_location/.zattrs": {
+            "_ARRAY_DIMENSIONS": [
+                "eta_rho",
+                "xi_rho"
+            ],
+            "long_name": "River index plus local volume fraction",
+            "units": "none"
+        },
         "river_name/.zarray": {
             "chunks": [
                 6

roms_tools/tests/test_setup/test_data/river_forcing_with_bgc.zarr/river_location/.zarray

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

roms_tools/tests/test_setup/test_data/river_forcing_with_bgc.zarr/river_location/.zattrs

@@ -0,0 +1,8 @@
+{
+    "_ARRAY_DIMENSIONS": [
+        "eta_rho",
+        "xi_rho"
+    ],
+    "long_name": "River index plus local volume fraction",
+    "units": "none"
+}

roms_tools/tests/test_setup/test_datasets.py

@@ -9,7 +9,7 @@ from roms_tools.setup.datasets import (
     ERA5Correction,
     CESMBGCDataset,
 )
-from roms_tools.setup.download import download_test_data
+from roms_tools.download import download_test_data
 from pathlib import Path
 
 

roms_tools/tests/test_setup/test_grid.py

@@ -4,7 +4,7 @@ import xarray as xr
 from roms_tools import Grid
 import importlib.metadata
 import textwrap
-from roms_tools.setup.download import download_test_data
+from roms_tools.download import download_test_data
 from conftest import calculate_file_hash
 from pathlib import Path
 
@@ -90,7 +90,6 @@ def test_plot_save_methods(tmp_path):
             str(tmp_path / file_str),
         ]:  # test for Path object and str
 
-            # Test saving without partitioning
             saved_filenames = grid.save(filepath)
             # Check if the .nc file was created
             filepath = Path(filepath).with_suffix(".nc")
@@ -100,18 +99,6 @@ def test_plot_save_methods(tmp_path):
             # Clean up the .nc file
             filepath.unlink()
 
-            # Test saving with partitioning
-            saved_filenames = grid.save(filepath, np_eta=2, np_xi=5)
-
-            filepath_str = str(filepath.with_suffix(""))
-            expected_filepath_list = [
-                Path(filepath_str + f".{index}.nc") for index in range(10)
-            ]
-            assert saved_filenames == expected_filepath_list
-            for expected_filepath in expected_filepath_list:
-                assert expected_filepath.exists()
-                expected_filepath.unlink()
-
 
 def test_raise_if_domain_too_large():
     with pytest.raises(ValueError, match="Domain size exceeds"):