black

Author: ashjbarnes

tests/conftest.py

@@ -114,7 +114,7 @@ def generate_silly_vt_dataset():
             ),
         }
     )
-    eastern_boundary.time.attrs = {"units":"days"}
+    eastern_boundary.time.attrs = {"units": "days"}
     return eastern_boundary
 
 
@@ -199,7 +199,6 @@ def dummy_bathymetry_data():
 def get_temperature_dataarrays(
     longitude_extent, latitude_extent, resolution, number_vertical_layers, depth
 ):
-
     silly_lat, silly_lon, silly_depth = generate_silly_coords(
         longitude_extent, latitude_extent, resolution, depth, number_vertical_layers
     )
@@ -217,9 +216,12 @@ def get_temperature_dataarrays(
 
     nx, ny = number_of_gridpoints(longitude_extent, latitude_extent, resolution)
 
-    temp_in_C, temp_in_C_masked, temp_in_K, temp_in_K_masked = (
-        generate_temperature_arrays(nx, ny, number_vertical_layers)
-    )
+    (
+        temp_in_C,
+        temp_in_C_masked,
+        temp_in_K,
+        temp_in_K_masked,
+    ) = generate_temperature_arrays(nx, ny, number_vertical_layers)
 
     temp_C = xr.DataArray(temp_in_C, dims=dims, coords=coords)
     temp_K = xr.DataArray(temp_in_K, dims=dims, coords=coords)
@@ -260,7 +262,6 @@ def generate_silly_coords(
 
 
 def generate_temperature_arrays(nx, ny, number_vertical_layers):
-
     # temperatures close to 0 ᵒC
     temp_in_C = np.random.randn(ny, nx, number_vertical_layers)
 

tests/test_regridding.py

@@ -23,11 +23,9 @@ def test_fill_missing_data(generate_silly_vt_dataset):
     """
     ds = generate_silly_vt_dataset
     ds["temp"][0, 0, 6:10, 0] = np.nan
-    ds.temp.attrs = {"units":"C"}
+    ds.temp.attrs = {"units": "C"}
     ds = rgd.fill_missing_data(ds, "silly_depth", fill="f")
-    assert (
-        ds.temp.attrs == {"units":"C"}
-    ) # Assert that attributes are retained
+    assert ds.temp.attrs == {"units": "C"}  # Assert that attributes are retained
     assert (
         ds["temp"][0, 0, 6:10, 0] == (ds["temp"][0, 0, 5, 0])
     ).all()  # Assert if we are forward filling in time
@@ -41,7 +39,7 @@ def test_add_or_update_time_dim(generate_silly_vt_dataset):
     ds = generate_silly_vt_dataset
 
     ds = rgd.add_or_update_time_dim(ds, xr.DataArray([0]))
-    assert ds.time.attrs == {"units":"days"} # Assert that attributes are retained
+    assert ds.time.attrs == {"units": "days"}  # Assert that attributes are retained
     assert ds["time"].values == [0]  # Assert time is added
     assert ds["temp"].dims[0] == "time"  # Check time is first dim
 
@@ -50,7 +48,7 @@ def test_generate_dz(generate_silly_vt_dataset):
     ds = generate_silly_vt_dataset
 
     dz = rgd.generate_dz(ds, "silly_depth")
-    assert ds.time.attrs == {"units":"days"} # Assert that attributes are retained
+    assert ds.time.attrs == {"units": "days"}  # Assert that attributes are retained
     z = np.linspace(0, 1000, 10)
     dz_check = np.full(z.shape, z[1] - z[0])
     assert (
@@ -62,11 +60,10 @@ def test_add_secondary_dimension(get_curvilinear_hgrid, generate_silly_vt_datase
     ds = generate_silly_vt_dataset
     hgrid = get_curvilinear_hgrid
 
-
     # N/S Boundary
     coords = rgd.coords(hgrid, "north", "segment_002")
     ds = rgd.add_secondary_dimension(ds, "temp", coords, "segment_002")
-    assert ds.time.attrs == {"units":"days"} # Assert that attributes are retained
+    assert ds.time.attrs == {"units": "days"}  # Assert that attributes are retained
     assert ds["temp"].dims == (
         "silly_lat",
         "ny_segment_002",
@@ -110,10 +107,8 @@ def test_add_secondary_dimension(get_curvilinear_hgrid, generate_silly_vt_datase
 def test_vertical_coordinate_encoding(generate_silly_vt_dataset):
     ds = generate_silly_vt_dataset
 
-
-
     ds = rgd.vertical_coordinate_encoding(ds, "temp", "segment_002", "silly_depth")
-    assert ds.time.attrs == {"units":"days"} # Assert that attributes are retained
+    assert ds.time.attrs == {"units": "days"}  # Assert that attributes are retained
 
     assert "nz_segment_002_temp" in ds["temp"].dims
     assert "nz_segment_002_temp" in ds
@@ -126,7 +121,7 @@ def test_generate_layer_thickness(generate_silly_vt_dataset):
     ds = generate_silly_vt_dataset
     ds["temp"] = ds["temp"].transpose("time", "silly_depth", "silly_lat", "silly_lon")
     ds = rgd.generate_layer_thickness(ds, "temp", "segment_002", "silly_depth")
-    assert ds.time.attrs == {"units":"days"} # Assert that attributes are retained
+    assert ds.time.attrs == {"units": "days"}  # Assert that attributes are retained
 
     assert "dz_temp" in ds
     assert ds["dz_temp"].dims == ("time", "nz_temp", "ny_segment_002", "nx_segment_002")
@@ -242,7 +237,7 @@ def test_mask_dataset(get_curvilinear_hgrid):
     ds["temp"] = ds["temp"].expand_dims("y", axis=0)
     ds["temp"] = ds["temp"].expand_dims("nz_temp", axis=0)
     ds["temp"] = ds["temp"].expand_dims("time", axis=0)
-    ds.temp.attrs = {"units":"C"}
+    ds.temp.attrs = {"units": "C"}
     fill_value = 36
     ds = rgd.mask_dataset(
         ds,
@@ -252,7 +247,7 @@ def test_mask_dataset(get_curvilinear_hgrid):
         x_dim_name="t_points_x",
         fill_value=fill_value,
     )
-    assert ds.temp.attrs == {"units":"C"}
+    assert ds.temp.attrs == {"units": "C"}
 
     assert (
         np.isnan(ds["temp"][0, 0, 0][start_ind * 2 + 2]) == False