Formatted Python code with black

DHI · Aug 3, 2022 · 9f0a922 · 9f0a922
1 parent 9e08774
commit 9f0a922
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/mikeio/spatial/grid_geometry.py b/mikeio/spatial/grid_geometry.py
@@ -650,7 +650,7 @@ def get_node_coordinates(self):
         gn = Grid2D(x=xn, y=yn)
         return gn.xy
 
-    def to_geometryFM(self,*,z=None, west=2, east=4, north=5,south=3):
+    def to_geometryFM(self, *, z=None, west=2, east=4, north=5, south=3):
         """convert Grid2D to GeometryFM
 
         Parameters
@@ -681,7 +681,7 @@ def to_geometryFM(self,*,z=None, west=2, east=4, north=5,south=3):
         zn = np.zeros_like(x)
         if z is not None:
             zn[:] = z
-        
+
         codes = np.zeros(n, dtype=int)
         codes[y == y[-1]] = north
         codes[x == x[-1]] = east

diff --git a/tests/test_geometry_grid.py b/tests/test_geometry_grid.py
@@ -268,13 +268,14 @@ def test_to_geometryFM():
     assert g.codes[-2] == 5  # north
     assert g.codes[-1] == 4  # east (upper right corner)
 
+
 def test_to_geometryFM_custom_z_custom_code():
     nx = 5
     ny = 3
     grd = Grid2D(nx=nx, dx=1, ny=ny, dy=2)
     g = grd.to_geometryFM(z=-12.0, west=30)
     assert isinstance(g, GeometryFM)
-    assert all(g.node_coordinates[:,2] == -12.0)
+    assert all(g.node_coordinates[:, 2] == -12.0)
 
     assert g.codes[0] == 30