roms-tools 1.4.1__py3-none-any.whl → 1.4.2__py3-none-any.whl

roms_tools/tests/test_setup/test_boundary_forcing.py

@@ -8,9 +8,7 @@ from pathlib import Path
 
 
 def test_boundary_forcing_creation(boundary_forcing):
-    """
-    Test the creation of the BoundaryForcing object.
-    """
+    """Test the creation of the BoundaryForcing object."""
 
     fname = download_test_data("GLORYS_coarse_test_data.nc")
 
@@ -43,9 +41,7 @@ def test_boundary_forcing_creation(boundary_forcing):
 
 
 def test_boundary_forcing_creation_with_bgc(bgc_boundary_forcing_from_climatology):
-    """
-    Test the creation of the BoundaryForcing object.
-    """
+    """Test the creation of the BoundaryForcing object."""
 
     fname_bgc = download_test_data("CESM_regional_coarse_test_data_climatology.nc")
 
@@ -80,9 +76,7 @@ def test_boundary_forcing_creation_with_bgc(bgc_boundary_forcing_from_climatolog
 
 
 def test_boundary_forcing_plot_save(boundary_forcing, tmp_path):
-    """
-    Test plot and save methods.
-    """
+    """Test plot and save methods."""
 
     boundary_forcing.plot(varname="temp_south", layer_contours=True)
     boundary_forcing.plot(varname="temp_east", layer_contours=True)
@@ -128,9 +122,7 @@ def test_boundary_forcing_plot_save(boundary_forcing, tmp_path):
 def test_bgc_boundary_forcing_plot_save(
     bgc_boundary_forcing_from_climatology, tmp_path
 ):
-    """
-    Test plot and save methods.
-    """
+    """Test plot and save methods."""
 
     bgc_boundary_forcing_from_climatology.plot(varname="ALK_south")
     bgc_boundary_forcing_from_climatology.plot(varname="ALK_east")
@@ -176,7 +168,8 @@ def test_bgc_boundary_forcing_plot_save(
     ],
 )
 def test_roundtrip_yaml(bdry_forcing_fixture, request, tmp_path, use_dask):
-    """Test that creating a BoundaryForcing object, saving its parameters to yaml file, and re-opening yaml file creates the same object."""
+    """Test that creating a BoundaryForcing object, saving its parameters to yaml file,
+    and re-opening yaml file creates the same object."""
 
     bdry_forcing = request.getfixturevalue(bdry_forcing_fixture)
 

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

@@ -1,10 +1,10 @@
 {
     "center_lat": 0,
     "center_lon": -20,
-    "coordinates": "interface_depth_rho interface_depth_u interface_depth_v lat_psi layer_depth_rho layer_depth_u layer_depth_v lon_psi",
+    "coordinates": "interface_depth_rho interface_depth_u interface_depth_v layer_depth_rho layer_depth_u layer_depth_v",
     "hc": 300.0,
     "hmin": 5.0,
-    "roms_tools_version": "0.1.dev138+dirty",
+    "roms_tools_version": "0.1.dev157+dirty",
     "rot": 0,
     "size_x": 100,
     "size_y": 100,

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

@@ -3,10 +3,10 @@
         ".zattrs": {
             "center_lat": 0,
             "center_lon": -20,
-            "coordinates": "interface_depth_rho interface_depth_u interface_depth_v lat_psi layer_depth_rho layer_depth_u layer_depth_v lon_psi",
+            "coordinates": "interface_depth_rho interface_depth_u interface_depth_v layer_depth_rho layer_depth_u layer_depth_v",
             "hc": 300.0,
             "hmin": 5.0,
-            "roms_tools_version": "0.1.dev138+dirty",
+            "roms_tools_version": "0.1.dev157+dirty",
             "rot": 0,
             "size_x": 100,
             "size_y": 100,
@@ -325,36 +325,6 @@
             "long_name": "latitude of rho-points on coarsened grid",
             "units": "degrees North"
         },
-        "lat_psi/.zarray": {
-            "chunks": [
-                4,
-                4
-            ],
-            "compressor": {
-                "blocksize": 0,
-                "clevel": 5,
-                "cname": "lz4",
-                "id": "blosc",
-                "shuffle": 1
-            },
-            "dtype": "<f8",
-            "fill_value": "NaN",
-            "filters": null,
-            "order": "C",
-            "shape": [
-                4,
-                4
-            ],
-            "zarr_format": 2
-        },
-        "lat_psi/.zattrs": {
-            "_ARRAY_DIMENSIONS": [
-                "eta_psi",
-                "xi_psi"
-            ],
-            "long_name": "latitude of psi-points",
-            "units": "degrees North"
-        },
         "lat_rho/.zarray": {
             "chunks": [
                 3,
@@ -574,36 +544,6 @@
             "long_name": "longitude of rho-points on coarsened grid",
             "units": "degrees East"
         },
-        "lon_psi/.zarray": {
-            "chunks": [
-                4,
-                4
-            ],
-            "compressor": {
-                "blocksize": 0,
-                "clevel": 5,
-                "cname": "lz4",
-                "id": "blosc",
-                "shuffle": 1
-            },
-            "dtype": "<f8",
-            "fill_value": "NaN",
-            "filters": null,
-            "order": "C",
-            "shape": [
-                4,
-                4
-            ],
-            "zarr_format": 2
-        },
-        "lon_psi/.zattrs": {
-            "_ARRAY_DIMENSIONS": [
-                "eta_psi",
-                "xi_psi"
-            ],
-            "long_name": "longitude of psi-points",
-            "units": "degrees East"
-        },
         "lon_rho/.zarray": {
             "chunks": [
                 3,

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

@@ -1,10 +1,10 @@
 {
     "center_lat": 0,
     "center_lon": 0,
-    "coordinates": "interface_depth_rho interface_depth_u interface_depth_v lat_psi layer_depth_rho layer_depth_u layer_depth_v lon_psi",
+    "coordinates": "interface_depth_rho interface_depth_u interface_depth_v layer_depth_rho layer_depth_u layer_depth_v",
     "hc": 300.0,
     "hmin": 5.0,
-    "roms_tools_version": "0.1.dev138+dirty",
+    "roms_tools_version": "0.1.dev157+dirty",
     "rot": 20,
     "size_x": 100,
     "size_y": 100,

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

@@ -3,10 +3,10 @@
         ".zattrs": {
             "center_lat": 0,
             "center_lon": 0,
-            "coordinates": "interface_depth_rho interface_depth_u interface_depth_v lat_psi layer_depth_rho layer_depth_u layer_depth_v lon_psi",
+            "coordinates": "interface_depth_rho interface_depth_u interface_depth_v layer_depth_rho layer_depth_u layer_depth_v",
             "hc": 300.0,
             "hmin": 5.0,
-            "roms_tools_version": "0.1.dev138+dirty",
+            "roms_tools_version": "0.1.dev157+dirty",
             "rot": 20,
             "size_x": 100,
             "size_y": 100,
@@ -325,36 +325,6 @@
             "long_name": "latitude of rho-points on coarsened grid",
             "units": "degrees North"
         },
-        "lat_psi/.zarray": {
-            "chunks": [
-                4,
-                4
-            ],
-            "compressor": {
-                "blocksize": 0,
-                "clevel": 5,
-                "cname": "lz4",
-                "id": "blosc",
-                "shuffle": 1
-            },
-            "dtype": "<f8",
-            "fill_value": "NaN",
-            "filters": null,
-            "order": "C",
-            "shape": [
-                4,
-                4
-            ],
-            "zarr_format": 2
-        },
-        "lat_psi/.zattrs": {
-            "_ARRAY_DIMENSIONS": [
-                "eta_psi",
-                "xi_psi"
-            ],
-            "long_name": "latitude of psi-points",
-            "units": "degrees North"
-        },
         "lat_rho/.zarray": {
             "chunks": [
                 3,
@@ -574,36 +544,6 @@
             "long_name": "longitude of rho-points on coarsened grid",
             "units": "degrees East"
         },
-        "lon_psi/.zarray": {
-            "chunks": [
-                4,
-                4
-            ],
-            "compressor": {
-                "blocksize": 0,
-                "clevel": 5,
-                "cname": "lz4",
-                "id": "blosc",
-                "shuffle": 1
-            },
-            "dtype": "<f8",
-            "fill_value": "NaN",
-            "filters": null,
-            "order": "C",
-            "shape": [
-                4,
-                4
-            ],
-            "zarr_format": 2
-        },
-        "lon_psi/.zattrs": {
-            "_ARRAY_DIMENSIONS": [
-                "eta_psi",
-                "xi_psi"
-            ],
-            "long_name": "longitude of psi-points",
-            "units": "degrees East"
-        },
         "lon_rho/.zarray": {
             "chunks": [
                 3,

roms_tools/tests/test_setup/test_datasets.py

@@ -125,9 +125,7 @@ def non_global_dataset():
     ],
 )
 def test_select_times(data_fixture, expected_time_values, request, tmp_path, use_dask):
-    """
-    Test selecting times with different datasets.
-    """
+    """Test selecting times with different datasets."""
     start_time = datetime(2022, 2, 1)
     end_time = datetime(2022, 3, 1)
 
@@ -160,9 +158,7 @@ def test_select_times(data_fixture, expected_time_values, request, tmp_path, use
 def test_select_times_valid_start_no_end_time(
     data_fixture, expected_time_values, request, tmp_path, use_dask
 ):
-    """
-    Test selecting times with only start_time specified.
-    """
+    """Test selecting times with only start_time specified."""
     start_time = datetime(2022, 2, 1)
 
     # Get the fixture dynamically based on the parameter
@@ -196,9 +192,7 @@ def test_select_times_valid_start_no_end_time(
 def test_select_times_invalid_start_no_end_time(
     data_fixture, expected_time_values, request, tmp_path, use_dask
 ):
-    """
-    Test selecting times with only start_time specified.
-    """
+    """Test selecting times with only start_time specified."""
     # Get the fixture dynamically based on the parameter
     dataset = request.getfixturevalue(data_fixture)
 
@@ -221,9 +215,8 @@ def test_select_times_invalid_start_no_end_time(
 def test_multiple_matching_times(
     global_dataset_with_multiple_times_per_day, tmp_path, use_dask
 ):
-    """
-    Test handling when multiple matching times are found when end_time is not specified.
-    """
+    """Test handling when multiple matching times are found when end_time is not
+    specified."""
     filepath = tmp_path / "test.nc"
     global_dataset_with_multiple_times_per_day.to_netcdf(filepath)
     dataset = Dataset(
@@ -237,9 +230,7 @@ def test_multiple_matching_times(
 
 
 def test_warnings_times(global_dataset, tmp_path, use_dask):
-    """
-    Test handling when no matching times are found.
-    """
+    """Test handling when no matching times are found."""
     # Create a temporary file
     filepath = tmp_path / "test.nc"
     global_dataset.to_netcdf(filepath)
@@ -271,9 +262,8 @@ def test_warnings_times(global_dataset, tmp_path, use_dask):
 def test_reverse_latitude_reverse_depth_choose_subdomain(
     global_dataset, tmp_path, use_dask
 ):
-    """
-    Test reversing latitude when it is not ascending, the choose_subdomain method, and the convert_to_negative_depth method of the Dataset class.
-    """
+    """Test reversing latitude when it is not ascending, the choose_subdomain method,
+    and the convert_to_negative_depth method of the Dataset class."""
     start_time = datetime(2022, 1, 1)
 
     filepath = tmp_path / "test.nc"

roms_tools/tests/test_setup/test_grid.py

@@ -45,16 +45,14 @@ def test_grid_creation(grid):
     ["grid", "counter_clockwise_rotated_grid", "clockwise_rotated_grid"],
 )
 def test_coords_relation(grid_fixture, request):
-    """
-    Test that the coordinates satisfy the expected relations on a C-grid.
-    """
+    """Test that the coordinates satisfy the expected relations on a C-grid."""
     grid = request.getfixturevalue(grid_fixture)
 
     # psi versus rho
-    assert grid.ds.lon_psi.min() < grid.ds.lon_rho.min()
-    assert grid.ds.lon_psi.max() > grid.ds.lon_rho.max()
-    assert grid.ds.lat_psi.min() < grid.ds.lat_rho.min()
-    assert grid.ds.lat_psi.max() > grid.ds.lat_rho.max()
+    # assert grid.ds.lon_psi.min() < grid.ds.lon_rho.min()
+    # assert grid.ds.lon_psi.max() > grid.ds.lon_rho.max()
+    # assert grid.ds.lat_psi.min() < grid.ds.lat_rho.min()
+    # assert grid.ds.lat_psi.max() > grid.ds.lat_rho.max()
 
     # Assertion with tolerance is necessary for non-rotated grids
     def assert_larger_equal_than_with_tolerance(value1, value2, tolerance=1e-5):
@@ -218,7 +216,8 @@ def test_compatability_with_matlab_grid(tmp_path):
 
 
 def test_roundtrip_netcdf(tmp_path):
-    """Test that creating a grid, saving it to file, and re-opening it is the same as just creating it."""
+    """Test that creating a grid, saving it to file, and re-opening it is the same as
+    just creating it."""
 
     # Initialize a Grid object using the initializer
     grid_init = Grid(
@@ -255,7 +254,8 @@ def test_roundtrip_netcdf(tmp_path):
 
 
 def test_roundtrip_yaml(tmp_path):
-    """Test that creating a grid, saving its parameters to yaml file, and re-opening yaml file creates the same grid."""
+    """Test that creating a grid, saving its parameters to yaml file, and re- opening
+    yaml file creates the same grid."""
 
     # Initialize a Grid object using the initializer
     grid_init = Grid(

roms_tools/tests/test_setup/test_initial_conditions.py

@@ -19,9 +19,7 @@ from conftest import calculate_file_hash
     ],
 )
 def test_initial_conditions_creation(ic_fixture, request):
-    """
-    Test the creation of the InitialConditions object.
-    """
+    """Test the creation of the InitialConditions object."""
 
     ic = request.getfixturevalue(ic_fixture)
 
@@ -181,9 +179,7 @@ def test_interpolation_from_climatology(
 def test_initial_conditions_plot_save(
     initial_conditions_with_bgc_from_climatology, tmp_path
 ):
-    """
-    Test plot and save methods.
-    """
+    """Test plot and save methods."""
 
     initial_conditions_with_bgc_from_climatology.plot(varname="temp", s=0)
     initial_conditions_with_bgc_from_climatology.plot(
@@ -251,7 +247,8 @@ def test_initial_conditions_plot_save(
 
 
 def test_roundtrip_yaml(initial_conditions, tmp_path, use_dask):
-    """Test that creating an InitialConditions object, saving its parameters to yaml file, and re-opening yaml file creates the same object."""
+    """Test that creating an InitialConditions object, saving its parameters to yaml
+    file, and re-opening yaml file creates the same object."""
 
     # Create a temporary filepath using the tmp_path fixture
     file_str = "test_yaml"

roms_tools/tests/test_setup/test_surface_forcing.py

@@ -9,9 +9,8 @@ from conftest import calculate_file_hash
 
 @pytest.fixture
 def grid_that_straddles_dateline():
-    """
-    Fixture for creating a domain that straddles the dateline and lies within the bounds of the regional ERA5 data.
-    """
+    """Fixture for creating a domain that straddles the dateline and lies within the
+    bounds of the regional ERA5 data."""
     grid = Grid(
         nx=20,
         ny=20,
@@ -27,8 +26,9 @@ def grid_that_straddles_dateline():
 
 @pytest.fixture
 def grid_that_straddles_dateline_but_is_too_big_for_regional_test_data():
-    """
-    Fixture for creating a domain that straddles the dateline but exceeds the bounds of the regional ERA5 data.
+    """Fixture for creating a domain that straddles the dateline but exceeds the bounds
+    of the regional ERA5 data.
+
     Centered east of dateline.
     """
     grid = Grid(
@@ -46,9 +46,11 @@ def grid_that_straddles_dateline_but_is_too_big_for_regional_test_data():
 
 @pytest.fixture
 def another_grid_that_straddles_dateline_but_is_too_big_for_regional_test_data():
-    """
-    Fixture for creating a domain that straddles the dateline but exceeds the bounds of the regional ERA5 data.
-    Centered west of dateline. This one was hard to catch for the nan_check for a long time, but should work now.
+    """Fixture for creating a domain that straddles the dateline but exceeds the bounds
+    of the regional ERA5 data.
+
+    Centered west of dateline. This one was hard to catch for the nan_check for a long
+    time, but should work now.
     """
     grid = Grid(
         nx=5,
@@ -65,9 +67,11 @@ def another_grid_that_straddles_dateline_but_is_too_big_for_regional_test_data()
 
 @pytest.fixture
 def grid_that_lies_east_of_dateline_less_than_five_degrees_away():
-    """
-    Fixture for creating a domain that lies east of Greenwich meridian, but less than 5 degrees away.
-    We care about the 5 degree mark because it decides whether the code handles the longitudes as straddling the dateline or not.
+    """Fixture for creating a domain that lies east of Greenwich meridian, but less than
+    5 degrees away.
+
+    We care about the 5 degree mark because it decides whether the code handles the
+    longitudes as straddling the dateline or not.
     """
 
     grid = Grid(
@@ -85,9 +89,11 @@ def grid_that_lies_east_of_dateline_less_than_five_degrees_away():
 
 @pytest.fixture
 def grid_that_lies_east_of_dateline_more_than_five_degrees_away():
-    """
-    Fixture for creating a domain that lies east of Greenwich meridian, more than 5 degrees away.
-    We care about the 5 degree mark because it decides whether the code handles the longitudes as straddling the dateline or not.
+    """Fixture for creating a domain that lies east of Greenwich meridian, more than 5
+    degrees away.
+
+    We care about the 5 degree mark because it decides whether the code handles the
+    longitudes as straddling the dateline or not.
     """
     grid = Grid(
         nx=5,
@@ -104,9 +110,11 @@ def grid_that_lies_east_of_dateline_more_than_five_degrees_away():
 
 @pytest.fixture
 def grid_that_lies_west_of_dateline_less_than_five_degrees_away():
-    """
-    Fixture for creating a domain that lies west of Greenwich meridian, less than 5 degrees away.
-    We care about the 5 degree mark because it decides whether the code handles the longitudes as straddling the dateline or not.
+    """Fixture for creating a domain that lies west of Greenwich meridian, less than 5
+    degrees away.
+
+    We care about the 5 degree mark because it decides whether the code handles the
+    longitudes as straddling the dateline or not.
     """
 
     grid = Grid(
@@ -124,9 +132,11 @@ def grid_that_lies_west_of_dateline_less_than_five_degrees_away():
 
 @pytest.fixture
 def grid_that_lies_west_of_dateline_more_than_five_degrees_away():
-    """
-    Fixture for creating a domain that lies west of Greenwich meridian, more than 5 degrees away.
-    We care about the 5 degree mark because it decides whether the code handles the longitudes as straddling the dateline or not.
+    """Fixture for creating a domain that lies west of Greenwich meridian, more than 5
+    degrees away.
+
+    We care about the 5 degree mark because it decides whether the code handles the
+    longitudes as straddling the dateline or not.
     """
 
     grid = Grid(
@@ -144,8 +154,10 @@ def grid_that_lies_west_of_dateline_more_than_five_degrees_away():
 
 @pytest.fixture
 def grid_that_straddles_180_degree_meridian():
-    """
-    Fixture for creating a domain that straddles 180 degree meridian. This is a good test grid for the global ERA5 data, which comes on an [-180, 180] longitude grid.
+    """Fixture for creating a domain that straddles 180 degree meridian.
+
+    This is a good test grid for the global ERA5 data, which comes on an [-180, 180]
+    longitude grid.
     """
 
     grid = Grid(
@@ -172,8 +184,7 @@ def grid_that_straddles_180_degree_meridian():
     ],
 )
 def test_successful_initialization_with_regional_data(grid_fixture, request, use_dask):
-    """
-    Test the initialization of SurfaceForcing with regional ERA5 data.
+    """Test the initialization of SurfaceForcing with regional ERA5 data.
 
     This test checks the following:
     1. SurfaceForcing object initializes successfully with provided regional data.
@@ -242,8 +253,7 @@ def test_successful_initialization_with_regional_data(grid_fixture, request, use
 def test_nan_detection_initialization_with_regional_data(
     grid_fixture, request, use_dask
 ):
-    """
-    Test handling of NaN values during initialization with regional data.
+    """Test handling of NaN values during initialization with regional data.
 
     Ensures ValueError is raised if NaN values are detected in the dataset.
     """
@@ -270,10 +280,11 @@ def test_nan_detection_initialization_with_regional_data(
 def test_no_longitude_intersection_initialization_with_regional_data(
     grid_that_straddles_180_degree_meridian, use_dask
 ):
-    """
-    Test initialization of SurfaceForcing with a grid that straddles the 180° meridian.
+    """Test initialization of SurfaceForcing with a grid that straddles the 180°
+    meridian.
 
-    Ensures ValueError is raised when the longitude range does not intersect with the dataset.
+    Ensures ValueError is raised when the longitude range does not intersect with the
+    dataset.
     """
     start_time = datetime(2020, 1, 31)
     end_time = datetime(2020, 2, 2)
@@ -309,11 +320,11 @@ def test_no_longitude_intersection_initialization_with_regional_data(
     ],
 )
 def test_successful_initialization_with_global_data(grid_fixture, request, use_dask):
-    """
-    Test initialization of SurfaceForcing with global data.
+    """Test initialization of SurfaceForcing with global data.
 
     Verifies that the SurfaceForcing object is correctly initialized with global data,
-    including the correct handling of the grid and physics data. Checks both coarse and fine grid initialization.
+    including the correct handling of the grid and physics data. Checks both coarse and
+    fine grid initialization.
     """
     start_time = datetime(2020, 1, 31)
     end_time = datetime(2020, 2, 2)
@@ -357,8 +368,7 @@ def test_successful_initialization_with_global_data(grid_fixture, request, use_d
 
 
 def test_nans_filled_in(grid_that_straddles_dateline, use_dask):
-    """
-    Test that the surface forcing fields contain no NaNs.
+    """Test that the surface forcing fields contain no NaNs.
 
     The test is performed twice:
     - First with the default fine grid.
@@ -405,10 +415,8 @@ def test_nans_filled_in(grid_that_straddles_dateline, use_dask):
 
 
 def test_time_attr_climatology(bgc_surface_forcing_from_climatology):
-    """
-    Test that the 'cycle_length' attribute is present in the time coordinate of the BGC dataset
-    when using climatology data.
-    """
+    """Test that the 'cycle_length' attribute is present in the time coordinate of the
+    BGC dataset when using climatology data."""
     for time_coord in ["pco2_time", "iron_time", "dust_time", "nox_time", "nhy_time"]:
         assert hasattr(
             bgc_surface_forcing_from_climatology.ds[time_coord],
@@ -418,10 +426,8 @@ def test_time_attr_climatology(bgc_surface_forcing_from_climatology):
 
 
 def test_time_attr(bgc_surface_forcing):
-    """
-    Test that the 'cycle_length' attribute is not present in the time coordinate of the BGC dataset
-    when not using climatology data.
-    """
+    """Test that the 'cycle_length' attribute is not present in the time coordinate of
+    the BGC dataset when not using climatology data."""
     for time_coord in ["pco2_time", "iron_time", "dust_time", "nox_time", "nhy_time"]:
         assert not hasattr(
             bgc_surface_forcing.ds[time_coord],
@@ -448,12 +454,11 @@ def test_time_attr(bgc_surface_forcing):
 def test_surface_forcing_creation(
     sfc_forcing_fixture, expected_climatology, expected_fname, request
 ):
-    """
-    Test the creation and initialization of the SurfaceForcing object with BGC.
+    """Test the creation and initialization of the SurfaceForcing object with BGC.
 
     Verifies that the SurfaceForcing object is properly created with correct attributes.
-    Ensures that expected variables are present in the dataset
-    and that attributes match the given configurations.
+    Ensures that expected variables are present in the dataset and that attributes match
+    the given configurations.
     """
 
     sfc_forcing = request.getfixturevalue(sfc_forcing_fixture)
@@ -491,9 +496,7 @@ def test_surface_forcing_creation(
     ],
 )
 def test_surface_forcing_plot_save(sfc_forcing_fixture, request, tmp_path):
-    """
-    Test plot and save methods.
-    """
+    """Test plot and save methods."""
     sfc_forcing = request.getfixturevalue(sfc_forcing_fixture)
     sfc_forcing.plot(varname="uwnd", time=0)
 
@@ -525,9 +528,7 @@ def test_surface_forcing_plot_save(sfc_forcing_fixture, request, tmp_path):
 
 
 def test_surface_forcing_bgc_plot_save(bgc_surface_forcing, tmp_path):
-    """
-    Test plot and save methods.
-    """
+    """Test plot and save methods."""
 
     # Check the values in the dataset
     bgc_surface_forcing.plot(varname="pco2_air", time=0)
@@ -562,9 +563,7 @@ def test_surface_forcing_bgc_plot_save(bgc_surface_forcing, tmp_path):
 def test_surface_forcing_bgc_from_clim_plot_save(
     bgc_surface_forcing_from_climatology, tmp_path
 ):
-    """
-    Test plot and save methods.
-    """
+    """Test plot and save methods."""
 
     # Check the values in the dataset
     bgc_surface_forcing_from_climatology.plot(varname="pco2_air", time=0)
@@ -609,7 +608,8 @@ def test_surface_forcing_bgc_from_clim_plot_save(
     ],
 )
 def test_roundtrip_yaml(sfc_forcing_fixture, request, tmp_path, use_dask):
-    """Test that creating an SurfaceForcing object, saving its parameters to yaml file, and re-opening yaml file creates the same object."""
+    """Test that creating an SurfaceForcing object, saving its parameters to yaml file,
+    and re-opening yaml file creates the same object."""
 
     sfc_forcing = request.getfixturevalue(sfc_forcing_fixture)