Rename method

Author: lauraporta

derotation/simulate/synthetic_data.py

@@ -124,7 +124,7 @@ def integration_pipeline(
 
         # -----------------------------------------------------
         # Create sample image with two cells
-        self.image = self.create_sample_image_with_two_cells()
+        self.image = self.create_sample_image_with_cells()
 
         # Create the 3D image stack
         image_stack = self.create_image_stack()
@@ -202,7 +202,7 @@ def integration_pipeline(
     #  Prepare the 3D image stack and the rotation angles
     #  -----------------------------------------------------
 
-    def create_sample_image_with_two_cells(self) -> np.ndarray:
+    def create_sample_image_with_cells(self) -> np.ndarray:
         """Create a 2D image with two circles, one bright and one dim
         (optional) by default in the top center and bottom right,
         respectively.
@@ -379,6 +379,7 @@ def __init__(self):
 
                 self.debugging_plots = make_plots
                 self.debug_plots_folder = Path("debug/")
+
             def calculate_mean_images(self, image_stack: np.ndarray) -> list:
                 #  Overwrite original method as it is too bound
                 #  to signal coming from a real motor

examples/elliptical_rotations.py

@@ -23,7 +23,7 @@ def rotate_image_stack(
         background_value=80,
         num_frames=50,
     )
-    s_data.image = s_data.create_sample_image_with_two_cells()
+    s_data.image = s_data.create_sample_image_with_cells()
     image_stack = s_data.create_image_stack()
     _, angles = s_data.create_rotation_angles(image_stack.shape)
 

tests/test_integration/test_rotation_out_of_plane.py

@@ -46,7 +46,7 @@ def rotate_image_stack(
         pad=pad,
         background_value=80,
     )
-    s_data.image = s_data.create_sample_image_with_two_cells()
+    s_data.image = s_data.create_sample_image_with_cells()
     image_stack = s_data.create_image_stack()
     _, angles = s_data.create_rotation_angles(image_stack.shape)