roms-tools 1.0.1__py3-none-any.whl → 1.2.0__py3-none-any.whl

roms_tools/tests/test_surface_forcing.py

@@ -193,8 +193,26 @@ def test_successful_initialization_with_regional_data(grid_fixture, request):
     )
 
     assert sfc_forcing.ds is not None
+    assert "uwnd" in sfc_forcing.ds["physics"]
+    assert "vwnd" in sfc_forcing.ds["physics"]
+    assert "swrad" in sfc_forcing.ds["physics"]
+    assert "lwrad" in sfc_forcing.ds["physics"]
+    assert "Tair" in sfc_forcing.ds["physics"]
+    assert "qair" in sfc_forcing.ds["physics"]
+    assert "rain" in sfc_forcing.ds["physics"]
+
+    assert sfc_forcing.start_time == start_time
+    assert sfc_forcing.end_time == end_time
+    assert sfc_forcing.physics_source == {
+        "name": "ERA5",
+        "path": fname,
+        "climatology": False,
+    }
+    assert not sfc_forcing.use_coarse_grid
 
-    grid.coarsen()
+    sfc_forcing.plot("uwnd", time=0)
+    sfc_forcing.plot("vwnd", time=0)
+    sfc_forcing.plot("rain", time=0)
 
     sfc_forcing = SurfaceForcing(
         grid=grid,
@@ -204,6 +222,7 @@ def test_successful_initialization_with_regional_data(grid_fixture, request):
         physics_source={"name": "ERA5", "path": fname},
     )
 
+    assert sfc_forcing.ds is not None
     assert "uwnd" in sfc_forcing.ds["physics"]
     assert "vwnd" in sfc_forcing.ds["physics"]
     assert "swrad" in sfc_forcing.ds["physics"]
@@ -219,6 +238,11 @@ def test_successful_initialization_with_regional_data(grid_fixture, request):
         "path": fname,
         "climatology": False,
     }
+    assert sfc_forcing.use_coarse_grid
+
+    sfc_forcing.plot("uwnd", time=0)
+    sfc_forcing.plot("vwnd", time=0)
+    sfc_forcing.plot("rain", time=0)
 
 
 @pytest.mark.parametrize(
@@ -250,8 +274,6 @@ def test_nan_detection_initialization_with_regional_data(grid_fixture, request):
             physics_source={"name": "ERA5", "path": fname},
         )
 
-    grid.coarsen()
-
     with pytest.raises(ValueError, match="NaN values found"):
         SurfaceForcing(
             grid=grid,
@@ -286,8 +308,6 @@ def test_no_longitude_intersection_initialization_with_regional_data(
             physics_source={"name": "ERA5", "path": fname},
         )
 
-    grid_that_straddles_180_degree_meridian.coarsen()
-
     with pytest.raises(
         ValueError, match="Selected longitude range does not intersect with dataset"
     ):
@@ -350,8 +370,6 @@ def test_successful_initialization_with_global_data(grid_fixture, request):
     assert "rain" in sfc_forcing.ds["physics"]
     assert sfc_forcing.ds["physics"].attrs["physics_source"] == "ERA5"
 
-    grid.coarsen()
-
     sfc_forcing = SurfaceForcing(
         grid=grid,
         use_coarse_grid=True,

roms_tools/tests/test_topography.py

@@ -31,12 +31,10 @@ def test_rmax_criterion():
         center_lon=30,
         center_lat=61,
         rot=20,
-        smooth_factor=4,
-        rmax=0.2,
     )
     r_eta, r_xi = _compute_rfactor(grid.ds.h)
     rmax0 = np.max([r_eta.max(), r_xi.max()])
-    npt.assert_array_less(rmax0, grid.rmax)
+    npt.assert_array_less(rmax0, 0.2)
 
 
 def test_hmin_criterion():
@@ -48,11 +46,15 @@ def test_hmin_criterion():
         center_lon=30,
         center_lat=61,
         rot=20,
-        smooth_factor=2,
-        rmax=0.2,
-        hmin=5,
+        hmin=5.0,
     )
 
+    assert grid.hmin == 5.0
+    assert np.less_equal(grid.hmin, grid.ds.h.min())
+
+    grid.update_topography_and_mask(hmin=10.0)
+
+    assert grid.hmin == 10.0
     assert np.less_equal(grid.hmin, grid.ds.h.min())
 
 

roms_tools/tests/test_vertical_coordinate.py

@@ -1,31 +1,22 @@
 import pytest
 import numpy as np
-import tempfile
-import os
-from roms_tools import Grid, VerticalCoordinate
-import textwrap
+from roms_tools import Grid
 
 
-@pytest.fixture
-def example_grid():
-    """
-    Fixture for creating a Grid object.
-    """
-    grid = Grid(
-        nx=2, ny=2, size_x=500, size_y=1000, center_lon=0, center_lat=55, rot=10
-    )
-
-    return grid
-
-
-def test_invalid_theta_s_value(example_grid):
+def test_invalid_theta_s_value():
     """
     Test the validation of the theta_s value.
     """
     with pytest.raises(ValueError):
 
-        VerticalCoordinate(
-            grid=example_grid,
+        Grid(
+            nx=2,
+            ny=2,
+            size_x=500,
+            size_y=1000,
+            center_lon=0,
+            center_lat=55,
+            rot=10,
             N=3,
             theta_s=11.0,  # Invalid value, should be 0 < theta_s <= 10
             theta_b=2.0,
@@ -33,14 +24,19 @@ def test_invalid_theta_s_value(example_grid):
         )
 
 
-def test_invalid_theta_b_value(example_grid):
+def test_invalid_theta_b_value():
     """
     Test the validation of the theta_b value.
     """
     with pytest.raises(ValueError):
-
-        VerticalCoordinate(
-            grid=example_grid,
+        Grid(
+            nx=2,
+            ny=2,
+            size_x=500,
+            size_y=1000,
+            center_lon=0,
+            center_lat=55,
+            rot=10,
             N=3,
             theta_s=5.0,
             theta_b=5.0,  # Invalid value, should be 0 < theta_b <= 4
@@ -48,47 +44,68 @@ def test_invalid_theta_b_value(example_grid):
         )
 
 
-@pytest.fixture
-def vertical_coordinate(example_grid):
+def test_update_vertical_coordinate():
 
-    return VerticalCoordinate(
-        grid=example_grid, N=3, theta_s=5.0, theta_b=2.0, hc=250.0
+    grid = Grid(
+        nx=2, ny=2, size_x=500, size_y=1000, center_lon=0, center_lat=55, rot=10
     )
 
+    assert grid.N == 100
+    assert grid.theta_s == 5.0
+    assert grid.theta_b == 2.0
+    assert grid.hc == 300.0
+    assert len(grid.ds.s_rho) == 100
+
+    grid.update_vertical_coordinate(N=3, theta_s=10.0, theta_b=1.0, hc=400.0)
+
+    assert grid.N == 3
+    assert grid.theta_s == 10.0
+    assert grid.theta_b == 1.0
+    assert grid.hc == 400.0
+    assert len(grid.ds.s_rho) == 3
+
 
-def test_vertical_coordinate_data_consistency(vertical_coordinate, tmp_path):
+def test_vertical_coordinate_data_consistency():
     """
     Test that the data within the VerticalCoordinate object remains consistent.
     """
 
+    grid = Grid(
+        nx=2,
+        ny=2,
+        size_x=500,
+        size_y=1000,
+        center_lon=0,
+        center_lat=55,
+        rot=10,
+        N=3,
+        theta_s=5.0,
+        theta_b=2.0,
+        hc=250.0,
+    )
+
     # Define the expected data
     expected_sc_r = np.array([-0.8333333, -0.5, -0.16666667], dtype=np.float32)
     expected_Cs_r = np.array([-0.6641397, -0.15129805, -0.01156188], dtype=np.float32)
     expected_layer_depth_rho = np.array(
         [
             [
-                [17.235603, 9.938617, 3.2495391],
-                [17.235603, 9.938617, 3.2495391],
-                [24.206884, 13.7451725, 4.4592304],
-                [24.206884, 13.7451725, 4.4592304],
+                [17.235603, 17.235603, 24.206884, 24.206884],
+                [17.235603, 17.235603, 24.206884, 24.206884],
+                [25.667452, 25.667452, 35.966946, 35.966946],
+                [25.667452, 25.667452, 35.966946, 35.966946],
             ],
             [
-                [17.235603, 9.938617, 3.2495391],
-                [17.235603, 9.938617, 3.2495391],
-                [24.206884, 13.7451725, 4.4592304],
-                [24.206884, 13.7451725, 4.4592304],
+                [9.938617, 9.938617, 13.7451725, 13.7451725],
+                [9.938617, 9.938617, 13.7451725, 13.7451725],
+                [14.528268, 14.528268, 19.916193, 19.916193],
+                [14.528268, 14.528268, 19.916193, 19.916193],
             ],
             [
-                [25.667452, 14.528268, 4.7055993],
-                [25.667452, 14.528268, 4.7055993],
-                [35.966946, 19.916193, 6.377065],
-                [35.966946, 19.916193, 6.377065],
-            ],
-            [
-                [25.667452, 14.528268, 4.7055993],
-                [25.667452, 14.528268, 4.7055993],
-                [35.966946, 19.916193, 6.377065],
-                [35.966946, 19.916193, 6.377065],
+                [3.2495391, 3.2495391, 4.4592304, 4.4592304],
+                [3.2495391, 3.2495391, 4.4592304, 4.4592304],
+                [4.7055993, 4.7055993, 6.377065, 6.377065],
+                [4.7055993, 4.7055993, 6.377065, 6.377065],
             ],
         ],
         dtype=np.float32,
@@ -96,24 +113,22 @@ def test_vertical_coordinate_data_consistency(vertical_coordinate, tmp_path):
     expected_layer_depth_u = np.array(
         [
             [
-                [17.235603, 9.938617, 3.2495391],
-                [20.721243, 11.841895, 3.854385],
-                [24.206884, 13.7451725, 4.4592304],
-            ],
-            [
-                [17.235603, 9.938617, 3.2495391],
-                [20.721243, 11.841895, 3.854385],
-                [24.206884, 13.7451725, 4.4592304],
+                [17.235603, 20.721243, 24.206884],
+                [17.235603, 20.721243, 24.206884],
+                [25.667452, 30.817198, 35.966946],
+                [25.667452, 30.817198, 35.966946],
             ],
             [
-                [25.667452, 14.528268, 4.7055993],
-                [30.817198, 17.22223, 5.5413322],
-                [35.966946, 19.916193, 6.377065],
+                [9.938617, 11.841895, 13.7451725],
+                [9.938617, 11.841895, 13.7451725],
+                [14.528268, 17.22223, 19.916193],
+                [14.528268, 17.22223, 19.916193],
             ],
             [
-                [25.667452, 14.528268, 4.7055993],
-                [30.817198, 17.22223, 5.5413322],
-                [35.966946, 19.916193, 6.377065],
+                [3.2495391, 3.854385, 4.4592304],
+                [3.2495391, 3.854385, 4.4592304],
+                [4.7055993, 5.5413322, 6.377065],
+                [4.7055993, 5.5413322, 6.377065],
             ],
         ],
         dtype=np.float32,
@@ -121,22 +136,19 @@ def test_vertical_coordinate_data_consistency(vertical_coordinate, tmp_path):
     expected_layer_depth_v = np.array(
         [
             [
-                [17.235603, 9.938617, 3.2495391],
-                [17.235603, 9.938617, 3.2495391],
-                [24.206884, 13.7451725, 4.4592304],
-                [24.206884, 13.7451725, 4.4592304],
+                [17.235603, 17.235603, 24.206884, 24.206884],
+                [21.451529, 21.451529, 30.086914, 30.086914],
+                [25.667452, 25.667452, 35.966946, 35.966946],
             ],
             [
-                [21.451529, 12.233442, 3.977569],
-                [21.451529, 12.233442, 3.977569],
-                [30.086914, 16.830683, 5.418148],
-                [30.086914, 16.830683, 5.418148],
+                [9.938617, 9.938617, 13.7451725, 13.7451725],
+                [12.233442, 12.233442, 16.830683, 16.830683],
+                [14.528268, 14.528268, 19.916193, 19.916193],
             ],
             [
-                [25.667452, 14.528268, 4.7055993],
-                [25.667452, 14.528268, 4.7055993],
-                [35.966946, 19.916193, 6.377065],
-                [35.966946, 19.916193, 6.377065],
+                [3.2495391, 3.2495391, 4.4592304, 4.4592304],
+                [3.977569, 3.977569, 5.418148, 5.418148],
+                [4.7055993, 4.7055993, 6.377065, 6.377065],
             ],
         ],
         dtype=np.float32,
@@ -144,28 +156,28 @@ def test_vertical_coordinate_data_consistency(vertical_coordinate, tmp_path):
     expected_interface_depth_rho = np.array(
         [
             [
-                [21.013529, 13.487298, 6.5489607, -0.0],
-                [21.013529, 13.487298, 6.5489607, -0.0],
-                [29.688076, 18.78298, 9.014939, -0.0],
-                [29.688076, 18.78298, 9.014939, -0.0],
+                [21.013529, 21.013529, 29.688076, 29.688076],
+                [21.013529, 21.013529, 29.688076, 29.688076],
+                [31.51735, 31.51735, 44.52708, 44.52708],
+                [31.51735, 31.51735, 44.52708, 44.52708],
             ],
             [
-                [21.013529, 13.487298, 6.5489607, -0.0],
-                [21.013529, 13.487298, 6.5489607, -0.0],
-                [29.688076, 18.78298, 9.014939, -0.0],
-                [29.688076, 18.78298, 9.014939, -0.0],
+                [13.487298, 13.487298, 18.78298, 18.78298],
+                [13.487298, 13.487298, 18.78298, 18.78298],
+                [19.881702, 19.881702, 27.529188, 27.529188],
+                [19.881702, 19.881702, 27.529188, 27.529188],
             ],
             [
-                [31.51735, 19.881702, 9.519225, -0.0],
-                [31.51735, 19.881702, 9.519225, -0.0],
-                [44.52708, 27.529188, 12.960222, -0.0],
-                [44.52708, 27.529188, 12.960222, -0.0],
+                [6.5489607, 6.5489607, 9.014939, 9.014939],
+                [6.5489607, 6.5489607, 9.014939, 9.014939],
+                [9.519225, 9.519225, 12.960222, 12.960222],
+                [9.519225, 9.519225, 12.960222, 12.960222],
             ],
             [
-                [31.51735, 19.881702, 9.519225, -0.0],
-                [31.51735, 19.881702, 9.519225, -0.0],
-                [44.52708, 27.529188, 12.960222, -0.0],
-                [44.52708, 27.529188, 12.960222, -0.0],
+                [-0.0, -0.0, -0.0, -0.0],
+                [-0.0, -0.0, -0.0, -0.0],
+                [-0.0, -0.0, -0.0, -0.0],
+                [-0.0, -0.0, -0.0, -0.0],
             ],
         ],
         dtype=np.float32,
@@ -173,24 +185,28 @@ def test_vertical_coordinate_data_consistency(vertical_coordinate, tmp_path):
     expected_interface_depth_u = np.array(
         [
             [
-                [21.013529, 13.487298, 6.5489607, -0.0],
-                [25.350803, 16.13514, 7.78195, -0.0],
-                [29.688076, 18.78298, 9.014939, -0.0],
+                [21.013529, 25.350803, 29.688076],
+                [21.013529, 25.350803, 29.688076],
+                [31.51735, 38.022217, 44.52708],
+                [31.51735, 38.022217, 44.52708],
             ],
             [
-                [21.013529, 13.487298, 6.5489607, -0.0],
-                [25.350803, 16.13514, 7.78195, -0.0],
-                [29.688076, 18.78298, 9.014939, -0.0],
+                [13.487298, 16.13514, 18.78298],
+                [13.487298, 16.13514, 18.78298],
+                [19.881702, 23.705446, 27.529188],
+                [19.881702, 23.705446, 27.529188],
             ],
             [
-                [31.51735, 19.881702, 9.519225, -0.0],
-                [38.022217, 23.705446, 11.239724, -0.0],
-                [44.52708, 27.529188, 12.960222, -0.0],
+                [6.5489607, 7.78195, 9.014939],
+                [6.5489607, 7.78195, 9.014939],
+                [9.519225, 11.239724, 12.960222],
+                [9.519225, 11.239724, 12.960222],
             ],
             [
-                [31.51735, 19.881702, 9.519225, -0.0],
-                [38.022217, 23.705446, 11.239724, -0.0],
-                [44.52708, 27.529188, 12.960222, -0.0],
+                [-0.0, -0.0, -0.0],
+                [-0.0, -0.0, -0.0],
+                [-0.0, -0.0, -0.0],
+                [-0.0, -0.0, -0.0],
             ],
         ],
         dtype=np.float32,
@@ -198,140 +214,69 @@ def test_vertical_coordinate_data_consistency(vertical_coordinate, tmp_path):
     expected_interface_depth_v = np.array(
         [
             [
-                [21.013529, 13.487298, 6.5489607, -0.0],
-                [21.013529, 13.487298, 6.5489607, -0.0],
-                [29.688076, 18.78298, 9.014939, -0.0],
-                [29.688076, 18.78298, 9.014939, -0.0],
+                [21.013529, 21.013529, 29.688076, 29.688076],
+                [26.26544, 26.26544, 37.10758, 37.10758],
+                [31.51735, 31.51735, 44.52708, 44.52708],
             ],
             [
-                [26.26544, 16.684502, 8.034093, -0.0],
-                [26.26544, 16.684502, 8.034093, -0.0],
-                [37.10758, 23.156084, 10.987581, -0.0],
-                [37.10758, 23.156084, 10.987581, -0.0],
+                [13.487298, 13.487298, 18.78298, 18.78298],
+                [16.684502, 16.684502, 23.156084, 23.156084],
+                [19.881702, 19.881702, 27.529188, 27.529188],
             ],
             [
-                [31.51735, 19.881702, 9.519225, -0.0],
-                [31.51735, 19.881702, 9.519225, -0.0],
-                [44.52708, 27.529188, 12.960222, -0.0],
-                [44.52708, 27.529188, 12.960222, -0.0],
+                [6.5489607, 6.5489607, 9.014939, 9.014939],
+                [8.034093, 8.034093, 10.987581, 10.987581],
+                [9.519225, 9.519225, 12.960222, 12.960222],
+            ],
+            [
+                [-0.0, -0.0, -0.0, -0.0],
+                [-0.0, -0.0, -0.0, -0.0],
+                [-0.0, -0.0, -0.0, -0.0],
             ],
         ],
         dtype=np.float32,
     )
+
     # Check the values in the dataset
-    assert np.allclose(vertical_coordinate.ds["sc_r"].values, expected_sc_r)
-    assert np.allclose(vertical_coordinate.ds["Cs_r"].values, expected_Cs_r)
-    assert np.allclose(
-        vertical_coordinate.ds["layer_depth_rho"].values, expected_layer_depth_rho
-    )
-    assert np.allclose(
-        vertical_coordinate.ds["layer_depth_u"].values, expected_layer_depth_u
-    )
+    assert np.allclose(grid.ds["sc_r"].values, expected_sc_r)
+    assert np.allclose(grid.ds["Cs_r"].values, expected_Cs_r)
+    assert np.allclose(grid.ds["layer_depth_rho"].values, expected_layer_depth_rho)
+    assert np.allclose(grid.ds["layer_depth_u"].values, expected_layer_depth_u)
+    assert np.allclose(grid.ds["layer_depth_v"].values, expected_layer_depth_v)
     assert np.allclose(
-        vertical_coordinate.ds["layer_depth_v"].values, expected_layer_depth_v
-    )
-    assert np.allclose(
-        vertical_coordinate.ds["interface_depth_rho"].values,
+        grid.ds["interface_depth_rho"].values,
         expected_interface_depth_rho,
     )
-    assert np.allclose(
-        vertical_coordinate.ds["interface_depth_u"].values, expected_interface_depth_u
-    )
-    assert np.allclose(
-        vertical_coordinate.ds["interface_depth_v"].values, expected_interface_depth_v
-    )
-
-
-def test_plot(vertical_coordinate):
-    vertical_coordinate.plot("layer_depth_u", s=0)
-    vertical_coordinate.plot("layer_depth_rho", s=-1)
-    vertical_coordinate.plot("interface_depth_v", s=-1)
-    vertical_coordinate.plot("layer_depth_rho", eta=0)
-    vertical_coordinate.plot("layer_depth_u", eta=0)
-    vertical_coordinate.plot("layer_depth_v", eta=0)
-    vertical_coordinate.plot("interface_depth_rho", eta=0)
-    vertical_coordinate.plot("interface_depth_u", eta=0)
-    vertical_coordinate.plot("interface_depth_v", eta=0)
-    vertical_coordinate.plot("layer_depth_rho", xi=0)
-    vertical_coordinate.plot("layer_depth_u", xi=0)
-    vertical_coordinate.plot("layer_depth_v", xi=0)
-    vertical_coordinate.plot("interface_depth_rho", xi=0)
-    vertical_coordinate.plot("interface_depth_u", xi=0)
-    vertical_coordinate.plot("interface_depth_v", xi=0)
-
-
-def test_save(vertical_coordinate, tmp_path):
-    filepath = tmp_path / "vertical_coordinate.nc"
-    vertical_coordinate.save(filepath)
-    assert filepath.exists()
-
-
-def test_roundtrip(vertical_coordinate):
-    """Test that creating a vertical_coordinate, saving it to file, and re-opening it is the same as just creating it."""
-
-    # Create a temporary file
-    with tempfile.NamedTemporaryFile(delete=False) as tmpfile:
-        filepath = tmpfile.name
-
-    try:
-        vertical_coordinate.save(filepath)
+    assert np.allclose(grid.ds["interface_depth_u"].values, expected_interface_depth_u)
+    assert np.allclose(grid.ds["interface_depth_v"].values, expected_interface_depth_v)
 
-        vertical_coordinate_from_file = VerticalCoordinate.from_file(filepath)
 
-        assert vertical_coordinate.ds == vertical_coordinate_from_file.ds
-
-    finally:
-        os.remove(filepath)
-
-
-def test_roundtrip_yaml(vertical_coordinate):
-    """Test that creating a VerticalCoordinate object, saving its parameters to yaml file, and re-opening yaml file creates the same object."""
-
-    # Create a temporary file
-    with tempfile.NamedTemporaryFile(delete=False) as tmpfile:
-        filepath = tmpfile.name
-
-    try:
-        vertical_coordinate.to_yaml(filepath)
-
-        vertical_coordinate_from_file = VerticalCoordinate.from_yaml(filepath)
-
-        assert vertical_coordinate == vertical_coordinate_from_file
-
-    finally:
-        os.remove(filepath)
-
-
-def test_from_yaml_missing_vertical_coordinate():
-    yaml_content = textwrap.dedent(
-        """\
-    ---
-    roms_tools_version: 0.0.0
-    ---
-    Grid:
-      nx: 100
-      ny: 100
-      size_x: 1800
-      size_y: 2400
-      center_lon: -10
-      center_lat: 61
-      rot: -20
-      topography_source: ETOPO5
-      smooth_factor: 8
-      hmin: 5.0
-      rmax: 0.2
-    """
+def test_plot():
+    grid = Grid(
+        nx=2,
+        ny=2,
+        size_x=500,
+        size_y=1000,
+        center_lon=0,
+        center_lat=55,
+        rot=10,
+        N=3,
+        theta_s=5.0,
+        theta_b=2.0,
+        hc=250.0,
     )
-
-    with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
-        yaml_filepath = tmp_file.name
-        tmp_file.write(yaml_content.encode())
-
-    try:
-        with pytest.raises(
-            ValueError,
-            match="No VerticalCoordinate configuration found in the YAML file.",
-        ):
-            VerticalCoordinate.from_yaml(yaml_filepath)
-    finally:
-        os.remove(yaml_filepath)
+    grid.plot_vertical_coordinate("layer_depth_u", s=0)
+    grid.plot_vertical_coordinate("layer_depth_rho", s=-1)
+    grid.plot_vertical_coordinate("interface_depth_v", s=-1)
+    grid.plot_vertical_coordinate("layer_depth_rho", eta=0)
+    grid.plot_vertical_coordinate("layer_depth_u", eta=0)
+    grid.plot_vertical_coordinate("layer_depth_v", eta=0)
+    grid.plot_vertical_coordinate("interface_depth_rho", eta=0)
+    grid.plot_vertical_coordinate("interface_depth_u", eta=0)
+    grid.plot_vertical_coordinate("interface_depth_v", eta=0)
+    grid.plot_vertical_coordinate("layer_depth_rho", xi=0)
+    grid.plot_vertical_coordinate("layer_depth_u", xi=0)
+    grid.plot_vertical_coordinate("layer_depth_v", xi=0)
+    grid.plot_vertical_coordinate("interface_depth_rho", xi=0)
+    grid.plot_vertical_coordinate("interface_depth_u", xi=0)
+    grid.plot_vertical_coordinate("interface_depth_v", xi=0)

{roms_tools-1.0.1.dist-info → roms_tools-1.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: roms-tools
-Version: 1.0.1
+Version: 1.2.0
 Summary: Tools for running and analysing UCLA-ROMS simulations
 Author-email: Nora Loose <nora.loose@gmail.com>, Thomas Nicholas <tom@cworthy.org>
 License: Apache-2
@@ -31,8 +31,7 @@ Requires-Dist: gcm-filters
 Requires-Dist: numba
 
 # ROMS-Tools
-
-[![PyPI version](https://badge.fury.io/py/roms-tools.svg)](https://badge.fury.io/py/roms-tools)
+[![PyPI version](https://img.shields.io/pypi/v/roms-tools.svg)](https://pypi.org/project/roms-tools/)
 [![codecov](https://codecov.io/gh/CWorthy-ocean/roms-tools/graph/badge.svg?token=5S1oNu39xE)](https://codecov.io/gh/CWorthy-ocean/roms-tools)
 [![Documentation Status](https://readthedocs.org/projects/roms-tools/badge/?version=latest)](https://roms-tools.readthedocs.io/en/latest/?badge=latest)
 ![Run Tests](https://github.com/CWorthy-ocean/roms-tools/actions/workflows/tests.yaml/badge.svg)