Change position of argument selection in and test.

MDAnalysis · Jul 7, 2021 · 6580a78 · 6580a78
1 parent 6546d1b
commit 6580a78
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/membrane_curvature/surface.py b/membrane_curvature/surface.py
@@ -1,7 +1,7 @@
 import numpy as np
 
 
-def derive_surface(n_cells_x, n_cells_y, selection, max_width_x, max_width_y):
+def derive_surface(selection, n_cells_x, n_cells_y, max_width_x, max_width_y):
     """
     Derive surface from AtomGroup positions.
 
@@ -105,6 +105,3 @@ def normalized_grid(grid_z_coordinates, grid_norm_unit):
     z_normalized = grid_z_coordinates / grid_norm_unit
 
     return z_normalized
-
-
-
diff --git a/membrane_curvature/tests/test_membrane_curvature.py b/membrane_curvature/tests/test_membrane_curvature.py
@@ -176,7 +176,7 @@ def test_derive_surface(small_grofile):
     n_cells, max_width = 3, 30
     expected_surface = np.array(([150., 150., 120.], [150., 120., 120.], [150., 120., 120.]))
     max_width_x = max_width_y = max_width
-    surface = derive_surface(n_cells, n_cells, small_grofile, max_width_x, max_width_y)
+    surface = derive_surface(small_grofile, n_cells, n_cells, max_width_x, max_width_y)
     assert_almost_equal(surface, expected_surface)