roms-tools 2.3.0__py3-none-any.whl → 2.5.0__py3-none-any.whl

roms_tools/setup/utils.py

@@ -3,7 +3,6 @@ import numpy as np
 from typing import Union, Any, Dict, Type
 import pandas as pd
 import cftime
-from roms_tools.utils import partition
 from pathlib import Path
 from datetime import datetime
 from dataclasses import fields, asdict
@@ -131,23 +130,36 @@ def interpolate_from_climatology(
     time_dim_name: str,
     time: Union[xr.DataArray, pd.DatetimeIndex],
 ) -> Union[xr.DataArray, xr.Dataset]:
-    """Interpolates the given field temporally based on the specified time points.
+    """Temporally interpolates a field based on specified time points.
 
-    If `field` is an xarray.Dataset, this function applies the interpolation to all data variables in the dataset.
+    This function performs temporal interpolation on the input `field` to match the provided `time` values.
+    If the input `field` is an `xarray.Dataset`, the interpolation is applied to all its data variables individually.
 
     Parameters
     ----------
     field : xarray.DataArray or xarray.Dataset
-        The field data to be interpolated. Can be a single DataArray or a Dataset.
+        The input field to be interpolated.
+        - If `field` is an `xarray.DataArray`, it should have a time dimension identified by `time_dim_name`.
+        - If `field` is an `xarray.Dataset`, all variables within the dataset are interpolated along the specified time dimension.
+        The time dimension is assumed to represent `day_of_year` for climatological purposes.
     time_dim_name : str
-        The name of the dimension in `field` that represents time.
+        The name of the time dimension in the `field`. This dimension is used for interpolation.
     time : xarray.DataArray or pandas.DatetimeIndex
-        The target time points for interpolation.
+        The target time points for interpolation. The time values should be compatible with the time format used in the `field`.
 
     Returns
     -------
     xarray.DataArray or xarray.Dataset
-        The field values interpolated to the specified time points. The type matches the input type.
+        The interpolated field, with the same type as the input (`xarray.DataArray` or `xarray.Dataset`),
+        but aligned to the specified `time` values.
+
+    Notes
+    -----
+    - The interpolation assumes the time dimension in `field` corresponds to `day_of_year`.
+      If the input time values are in a datetime format, ensure they are converted to `day_of_year` before calling this function.
+      For example, you can preprocess the time as follows:
+
+      >>> field["time"] = field["time"].dt.dayofyear
     """
 
     def interpolate_single_field(data_array: xr.DataArray) -> xr.DataArray:
@@ -157,11 +169,11 @@ def interpolate_from_climatology(
             day_of_year = time.dt.dayofyear
         else:
             if np.size(time) == 1:
-                day_of_year = time.timetuple().tm_yday
+                # Convert single datetime64 object to pandas.Timestamp
+                day_of_year = pd.Timestamp(time).dayofyear
             else:
-                day_of_year = np.array([t.timetuple().tm_yday for t in time])
-
-        data_array[time_dim_name] = data_array[time_dim_name].dt.days
+                # Convert each datetime64 object in the array to pandas.Timestamp
+                day_of_year = np.array([pd.Timestamp(t).dayofyear for t in time])
 
         # Concatenate across the beginning and end of the year
         time_concat = xr.concat(
@@ -201,6 +213,7 @@ def interpolate_from_climatology(
             for var, data_array in field.data_vars.items()
         }
         return xr.Dataset(interpolated_data_vars, attrs=field.attrs)
+
     else:
         raise TypeError("Input 'field' must be an xarray.DataArray or xarray.Dataset.")
 
@@ -568,59 +581,6 @@ def group_by_year(ds, filepath):
     return dataset_list, output_filenames
 
 
-def save_datasets(dataset_list, output_filenames, np_eta=None, np_xi=None):
-    """Save the list of datasets to netCDF4 files, with optional spatial partitioning.
-
-    Parameters
-    ----------
-    dataset_list : list
-        List of datasets to be saved.
-    output_filenames : list
-        List of filenames for the output files.
-    np_eta : int, optional
-        The number of partitions along the `eta` direction. If `None`, no spatial partitioning is performed.
-    np_xi : int, optional
-        The number of partitions along the `xi` direction. If `None`, no spatial partitioning is performed.
-
-    Returns
-    -------
-    List[Path]
-        A list of Path objects for the filenames that were saved.
-    """
-
-    saved_filenames = []
-
-    if np_eta is None and np_xi is None:
-        # Save the dataset as a single file
-        output_filenames = [f"{filename}.nc" for filename in output_filenames]
-        xr.save_mfdataset(dataset_list, output_filenames)
-
-        saved_filenames.extend(Path(f) for f in output_filenames)
-
-    else:
-        # Partition the dataset and save each partition as a separate file
-        np_eta = np_eta or 1
-        np_xi = np_xi or 1
-
-        partitioned_datasets = []
-        partitioned_filenames = []
-        for dataset, base_filename in zip(dataset_list, output_filenames):
-            partition_indices, partitions = partition(
-                dataset, np_eta=np_eta, np_xi=np_xi
-            )
-            partition_filenames = [
-                f"{base_filename}.{index}.nc" for index in partition_indices
-            ]
-            partitioned_datasets.extend(partitions)
-            partitioned_filenames.extend(partition_filenames)
-
-        xr.save_mfdataset(partitioned_datasets, partitioned_filenames)
-
-        saved_filenames.extend(Path(f) for f in partitioned_filenames)
-
-    return saved_filenames
-
-
 def get_target_coords(grid, use_coarse_grid=False):
     """Retrieves longitude and latitude coordinates from the grid, adjusting them based
     on longitude range.
@@ -652,8 +612,6 @@ def get_target_coords(grid, use_coarse_grid=False):
         mask = grid.ds.get("mask_coarse")
         if mask is not None:
             mask = mask.rename({"eta_coarse": "eta_rho", "xi_coarse": "xi_rho"})
-            mask_u = interpolate_from_rho_to_u(mask, method="multiplicative")
-            mask_v = interpolate_from_rho_to_v(mask, method="multiplicative")
 
         lat_psi = grid.ds.get("lat_psi_coarse")
         lon_psi = grid.ds.get("lon_psi_coarse")
@@ -663,8 +621,6 @@ def get_target_coords(grid, use_coarse_grid=False):
         lon = grid.ds.lon_rho
         angle = grid.ds.angle
         mask = grid.ds.get("mask_rho")
-        mask_u = grid.ds.get("mask_u")
-        mask_v = grid.ds.get("mask_v")
         lat_psi = grid.ds.get("lat_psi")
         lon_psi = grid.ds.get("lon_psi")
 
@@ -687,8 +643,6 @@ def get_target_coords(grid, use_coarse_grid=False):
         "lon_psi": lon_psi,
         "angle": angle,
         "mask": mask,
-        "mask_u": mask_u,
-        "mask_v": mask_v,
         "straddle": straddle,
     }
 
@@ -939,13 +893,9 @@ def _to_yaml(forcing_object, filepath: Union[str, Path]) -> None:
     if hasattr(forcing_object, "grid") and forcing_object.grid is not None:
         grid_data = asdict(forcing_object.grid)
         grid_yaml_data = {"Grid": _pop_grid_data(grid_data)}
-    else:
-        parent_grid_data = asdict(forcing_object.parent_grid)
-        parent_grid_yaml_data = {"ParentGrid": _pop_grid_data(parent_grid_data)}
-        child_grid_data = asdict(forcing_object.child_grid)
-        child_grid_yaml_data = {"ChildGrid": _pop_grid_data(child_grid_data)}
-
-        grid_yaml_data = {**parent_grid_yaml_data, **child_grid_yaml_data}
+    elif hasattr(forcing_object, "parent_grid"):
+        grid_data = asdict(forcing_object.parent_grid)
+        grid_yaml_data = {"ParentGrid": _pop_grid_data(grid_data)}
 
     # Step 2: Ensure Paths are Strings
     def ensure_paths_are_strings(obj, key):
@@ -979,11 +929,12 @@ def _to_yaml(forcing_object, filepath: Union[str, Path]) -> None:
         not in (
             "grid",
             "parent_grid",
-            "child_grid",
             "ds",
             "use_dask",
-            "bypass_validation",
             "climatology",
+            "verbose",
+            "straddle",
+            "indices",
         )
     ]
 
@@ -998,18 +949,37 @@ def _to_yaml(forcing_object, filepath: Union[str, Path]) -> None:
         # Add the field and its value to the forcing_data dictionary
         forcing_data[field_name] = value
 
-    # Step 5: Combine Grid and Forcing Data into a single dictionary for the final YAML content
+    # Step 5: Serialize `indices` data (conditionally)
+    indices_data = getattr(forcing_object, "indices", None)
+    if indices_data is not None:
+        serialized_indices = {
+            "_convention": "eta_rho, xi_rho"
+        }  # Add convention metadata
+
+        for key, value in indices_data.items():
+            serialized_indices[key] = [
+                f"{tup[0]}, {tup[1]}" for tup in value
+            ]  # Comma-separated string
+
+        forcing_data["indices"] = serialized_indices
+
+    # Step 6: Combine Grid and Forcing Data into a single dictionary for the final YAML content
     yaml_data = {
         **grid_yaml_data,  # Add the grid data to the final YAML structure
         forcing_object.__class__.__name__: forcing_data,  # Include the serialized forcing object data
     }
 
-    # Step 6: Write to YAML file
+    # Step 7: Write to YAML file
     with filepath.open("w") as file:
         # Write the header first
         file.write(header)
         # Write the serialized YAML data
-        yaml.dump(yaml_data, file, default_flow_style=False, sort_keys=False)
+        yaml.dump(
+            yaml_data,
+            file,
+            default_flow_style=False,
+            sort_keys=False,
+        )
 
 
 def _pop_grid_data(grid_data):

roms_tools/tests/test_analysis/test_roms_output.py

@@ -18,7 +18,6 @@ def roms_output_from_restart_file(use_dask):
     return ROMSOutput(
         grid=grid,
         path=Path(download_test_data("eastpac25km_rst.19980106000000.nc")),
-        type="restart",
         use_dask=use_dask,
     )
 
@@ -28,20 +27,6 @@ def test_load_model_output_file(roms_output_from_restart_file, use_dask):
     assert isinstance(roms_output_from_restart_file.ds, xr.Dataset)
 
 
-def test_load_model_output_directory(use_dask):
-    fname_grid = Path(download_test_data("epac25km_grd.nc"))
-    grid = Grid.from_file(fname_grid)
-
-    # Download at least two files, so these will be found within the pooch directory
-    _ = Path(download_test_data("eastpac25km_rst.19980106000000.nc"))
-    _ = Path(download_test_data("eastpac25km_rst.19980126000000.nc"))
-
-    # Directory
-    directory = os.path.dirname(download_test_data("eastpac25km_rst.19980106000000.nc"))
-    output = ROMSOutput(grid=grid, path=directory, type="restart", use_dask=use_dask)
-    assert isinstance(output.ds, xr.Dataset)
-
-
 def test_load_model_output_file_list(use_dask):
     fname_grid = Path(download_test_data("epac25km_grd.nc"))
     grid = Grid.from_file(fname_grid)
@@ -49,24 +34,23 @@ def test_load_model_output_file_list(use_dask):
     # List of files
     file1 = Path(download_test_data("eastpac25km_rst.19980106000000.nc"))
     file2 = Path(download_test_data("eastpac25km_rst.19980126000000.nc"))
-    output = ROMSOutput(
-        grid=grid, path=[file1, file2], type="restart", use_dask=use_dask
-    )
+    output = ROMSOutput(grid=grid, path=[file1, file2], use_dask=use_dask)
     assert isinstance(output.ds, xr.Dataset)
 
 
-def test_invalid_type(use_dask):
+def test_load_model_output_with_wildcard(use_dask):
     fname_grid = Path(download_test_data("epac25km_grd.nc"))
     grid = Grid.from_file(fname_grid)
 
-    # Invalid type
-    with pytest.raises(ValueError, match="Invalid type 'invalid_type'"):
-        ROMSOutput(
-            grid=grid,
-            path=Path(download_test_data("eastpac25km_rst.19980106000000.nc")),
-            type="invalid_type",
-            use_dask=use_dask,
-        )
+    # Download at least two files, so these will be found within the pooch directory
+    Path(download_test_data("eastpac25km_rst.19980106000000.nc"))
+    Path(download_test_data("eastpac25km_rst.19980126000000.nc"))
+    directory = Path(
+        os.path.dirname(download_test_data("eastpac25km_rst.19980106000000.nc"))
+    )
+
+    output = ROMSOutput(grid=grid, path=directory / "*rst*.nc", use_dask=use_dask)
+    assert isinstance(output.ds, xr.Dataset)
 
 
 def test_invalid_path(use_dask):
@@ -78,16 +62,6 @@ def test_invalid_path(use_dask):
         ROMSOutput(
             grid=grid,
             path=Path("/path/to/nonexistent/file.nc"),
-            type="restart",
-            use_dask=use_dask,
-        )
-
-    # Non-existent directory
-    with pytest.raises(FileNotFoundError):
-        ROMSOutput(
-            grid=grid,
-            path=Path("/path/to/nonexistent/directory"),
-            type="restart",
             use_dask=use_dask,
         )
 
@@ -99,7 +73,6 @@ def test_set_correct_model_reference_date(use_dask):
     output = ROMSOutput(
         grid=grid,
         path=Path(download_test_data("eastpac25km_rst.19980106000000.nc")),
-        type="restart",
         use_dask=use_dask,
     )
     assert output.model_reference_date == datetime(1995, 1, 1)
@@ -117,7 +90,6 @@ def test_model_reference_date_mismatch(use_dask):
         ROMSOutput(
             grid=grid,
             path=Path(download_test_data("eastpac25km_rst.19980106000000.nc")),
-            type="restart",
             model_reference_date=model_ref_date,
             use_dask=use_dask,
         )
@@ -138,7 +110,7 @@ def test_model_reference_date_no_metadata(use_dask, tmp_path, caplog):
             expected_exception,
             match="Model reference date could not be inferred from the metadata",
         ):
-            ROMSOutput(grid=grid, path=fname_mod, type="restart", use_dask=use_dask)
+            ROMSOutput(grid=grid, path=fname_mod, use_dask=use_dask)
 
         # Test case 2: When a model reference date is explicitly set, verify the warning
         with caplog.at_level(logging.WARNING):
@@ -146,7 +118,6 @@ def test_model_reference_date_no_metadata(use_dask, tmp_path, caplog):
                 grid=grid,
                 path=fname_mod,
                 model_reference_date=datetime(1995, 1, 1),
-                type="restart",
                 use_dask=use_dask,
             )
 
@@ -175,20 +146,53 @@ def test_model_reference_date_no_metadata(use_dask, tmp_path, caplog):
 def test_compute_depth_coordinates(use_dask):
     fname_grid = Path(download_test_data("epac25km_grd.nc"))
     grid = Grid.from_file(fname_grid)
+    fname_restart1 = Path(download_test_data("eastpac25km_rst.19980106000000.nc"))
+
+    for adjust_depth_for_sea_surface_height in [True, False]:
+        output = ROMSOutput(
+            grid=grid,
+            path=fname_restart1,
+            use_dask=use_dask,
+            adjust_depth_for_sea_surface_height=adjust_depth_for_sea_surface_height,
+        )
+
+        # Before calling get_vertical_coordinates, check if the dataset doesn't already have depth coordinates
+        assert "layer_depth_rho" not in output.ds_depth_coords.data_vars
+
+        # Call the method to get vertical coordinates
+        output._get_depth_coordinates(depth_type="layer")
+
+        # Check if the depth coordinates were added
+        assert "layer_depth_rho" in output.ds_depth_coords.data_vars
+
+
+def test_missing_zeta_gets_raised(use_dask):
+    """Test that a ValueError is raised when `zeta` is missing from the dataset and
+    `adjust_depth_for_sea_surface_height` is enabled."""
+    # Load the grid
+    fname_grid = Path(download_test_data("epac25km_grd.nc"))
+    grid = Grid.from_file(fname_grid)
 
+    # Load the ROMS output
     fname_restart1 = Path(download_test_data("eastpac25km_rst.19980106000000.nc"))
-    output = ROMSOutput(
-        grid=grid, path=fname_restart1, type="restart", use_dask=use_dask
+    roms_output = ROMSOutput(
+        grid=grid,
+        path=fname_restart1,
+        use_dask=use_dask,
+        adjust_depth_for_sea_surface_height=True,
     )
 
-    # Before calling get_vertical_coordinates, check if the dataset doesn't already have depth coordinates
-    assert "layer_depth_rho" not in output.ds.data_vars
-
-    # Call the method to get vertical coordinates
-    output.compute_depth_coordinates(depth_type="layer")
+    # Remove `zeta` from the dataset
+    object.__setattr__(
+        roms_output, "ds", roms_output.ds.drop_vars("zeta", errors="ignore")
+    )
 
-    # Check if the depth coordinates were added
-    assert "layer_depth_rho" in output.ds.data_vars
+    # Expect ValueError when calling `_get_depth_coordinates`
+    with pytest.raises(
+        ValueError,
+        match="`zeta` is required in provided ROMS output when `adjust_depth_for_sea_surface_height` is enabled.",
+    ):
+        roms_output._get_depth_coordinates()
 
 
 def test_check_vertical_coordinate_mismatch(use_dask):
@@ -196,9 +200,7 @@ def test_check_vertical_coordinate_mismatch(use_dask):
     grid = Grid.from_file(fname_grid)
 
     fname_restart1 = Path(download_test_data("eastpac25km_rst.19980106000000.nc"))
-    output = ROMSOutput(
-        grid=grid, path=fname_restart1, type="restart", use_dask=use_dask
-    )
+    output = ROMSOutput(grid=grid, path=fname_restart1, use_dask=use_dask)
 
     # create a mock dataset with inconsistent vertical coordinate parameters
     ds_mock = output.ds.copy()
@@ -226,9 +228,7 @@ def test_that_coordinates_are_added(use_dask):
     grid = Grid.from_file(fname_grid)
 
     fname_restart1 = Path(download_test_data("eastpac25km_rst.19980106000000.nc"))
-    output = ROMSOutput(
-        grid=grid, path=fname_restart1, type="restart", use_dask=use_dask
-    )
+    output = ROMSOutput(grid=grid, path=fname_restart1, use_dask=use_dask)
 
     assert "abs_time" in output.ds.coords
     assert "lat_rho" in output.ds.coords
@@ -239,23 +239,186 @@ def test_plot(roms_output_from_restart_file, use_dask):
 
     kwargs = {}
     for var_name in ["temp", "u", "v"]:
-        roms_output_from_restart_file.plot(var_name, time=0, s=-1, **kwargs)
-        roms_output_from_restart_file.plot(var_name, time=0, eta=0, **kwargs)
-        roms_output_from_restart_file.plot(var_name, time=0, xi=0, **kwargs)
-        roms_output_from_restart_file.plot(var_name, time=0, eta=0, xi=0, **kwargs)
-        roms_output_from_restart_file.plot(var_name, time=0, s=-1, eta=0, **kwargs)
+        for include_boundary in [False, True]:
+            roms_output_from_restart_file.plot(
+                var_name, time=0, s=-1, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name, time=0, eta=0, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name, time=0, eta=1, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name, time=0, xi=0, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name, time=0, xi=1, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                eta=0,
+                xi=0,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                eta=0,
+                xi=1,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                eta=1,
+                xi=0,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                eta=1,
+                xi=1,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                s=-1,
+                eta=0,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                s=-1,
+                eta=1,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                s=-1,
+                xi=0,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                s=-1,
+                xi=1,
+                **kwargs,
+                include_boundary=include_boundary
+            )
 
     kwargs = {"depth_contours": True, "layer_contours": True}
     for var_name in ["temp", "u", "v"]:
-        roms_output_from_restart_file.plot(var_name, time=0, s=-1, **kwargs)
-        roms_output_from_restart_file.plot(var_name, time=0, eta=0, **kwargs)
-        roms_output_from_restart_file.plot(var_name, time=0, xi=0, **kwargs)
-        roms_output_from_restart_file.plot(var_name, time=0, eta=0, xi=0, **kwargs)
-        roms_output_from_restart_file.plot(var_name, time=0, s=-1, eta=0, **kwargs)
-
-    roms_output_from_restart_file.plot("zeta", time=0, **kwargs)
-    roms_output_from_restart_file.plot("zeta", time=0, eta=0, **kwargs)
-    roms_output_from_restart_file.plot("zeta", time=0, xi=0, **kwargs)
+        for include_boundary in [False, True]:
+            roms_output_from_restart_file.plot(
+                var_name, time=0, s=-1, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name, time=0, eta=0, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name, time=0, eta=1, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name, time=0, xi=0, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name, time=0, xi=1, **kwargs, include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                eta=0,
+                xi=0,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                eta=0,
+                xi=1,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                eta=1,
+                xi=0,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                eta=1,
+                xi=1,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                s=-1,
+                eta=0,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                s=-1,
+                eta=1,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                s=-1,
+                xi=0,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+            roms_output_from_restart_file.plot(
+                var_name,
+                time=0,
+                s=-1,
+                xi=1,
+                **kwargs,
+                include_boundary=include_boundary
+            )
+
+    for include_boundary in [False, True]:
+        roms_output_from_restart_file.plot(
+            "zeta", time=0, **kwargs, include_boundary=include_boundary
+        )
+        roms_output_from_restart_file.plot(
+            "zeta", time=0, eta=0, **kwargs, include_boundary=include_boundary
+        )
+        roms_output_from_restart_file.plot(
+            "zeta", time=0, eta=1, **kwargs, include_boundary=include_boundary
+        )
+        roms_output_from_restart_file.plot(
+            "zeta", time=0, xi=0, **kwargs, include_boundary=include_boundary
+        )
+        roms_output_from_restart_file.plot(
+            "zeta", time=0, xi=1, **kwargs, include_boundary=include_boundary
+        )
 
 
 def test_plot_errors(roms_output_from_restart_file, use_dask):