all unit tests

Author: Jammy2211

autoarray/inversion/mock/mock_image_mesh.py

@@ -12,15 +12,3 @@ def __init__(self, image_plane_mesh_grid=None):
         super().__init__()
 
         self.image_plane_mesh_grid = image_plane_mesh_grid
-
-    def image_plane_mesh_grid_from(
-        self, mask: Mask2D, adapt_data: Optional[np.ndarray],
-    ) -> Grid2DIrregular:
-        if adapt_data is not None and self.image_plane_mesh_grid is not None:
-            return adapt_data * self.image_plane_mesh_grid
-
-        return self.image_plane_mesh_grid
-
-    @property
-    def uses_adapt_images(self) -> bool:
-        return False

autoarray/inversion/pixelization/image_mesh/hilbert.py

@@ -288,9 +288,7 @@ def image_plane_mesh_grid_from(
             gridy=grid_hb[:, 0],
         )
 
-        mesh_grid = Grid2DIrregular(values=np.stack((drawn_y, drawn_x), axis=-1))
-
-        return mesh_grid
+        return Grid2DIrregular(values=np.stack((drawn_y, drawn_x), axis=-1))
 
     def check_mesh_pixels_per_image_pixels(
         self,

autoarray/preloads.py

@@ -107,7 +107,9 @@ def __init__(
             for image_plane_mesh_grid in image_plane_mesh_grid_list:
 
                 if image_plane_mesh_grid is not None:
-                    self.image_plane_mesh_grid_list.append(np.array(image_plane_mesh_grid))
+                    self.image_plane_mesh_grid_list.append(
+                        np.array(image_plane_mesh_grid)
+                    )
                 else:
                     self.image_plane_mesh_grid_list.append(None)
 

test_autoarray/inversion/pixelization/image_mesh/test_abstract.py

@@ -44,28 +44,31 @@ def test__mesh_pixels_per_image_pixels_from(mask, mesh_grid, image_mesh):
 
 def test__check_mesh_pixels_per_image_pixels(mask, mesh_grid, image_mesh):
     image_mesh.check_mesh_pixels_per_image_pixels(
-        mask=mask, mesh_grid=mesh_grid,
+        mask=mask,
+        mesh_grid=mesh_grid,
     )
 
     image_mesh.check_mesh_pixels_per_image_pixels(
         mask=mask,
         mesh_grid=mesh_grid,
         image_mesh_min_mesh_pixels_per_pixel=3,
-        image_mesh_min_mesh_number=1
+        image_mesh_min_mesh_number=1,
     )
 
     with pytest.raises(aa.exc.InversionException):
         image_mesh.check_mesh_pixels_per_image_pixels(
             mask=mask,
             mesh_grid=mesh_grid,
-            image_mesh_min_mesh_pixels_per_pixel=5, image_mesh_min_mesh_number=1
+            image_mesh_min_mesh_pixels_per_pixel=5,
+            image_mesh_min_mesh_number=1,
         )
 
     with pytest.raises(aa.exc.InversionException):
         image_mesh.check_mesh_pixels_per_image_pixels(
             mask=mask,
             mesh_grid=mesh_grid,
-            image_mesh_min_mesh_pixels_per_pixel=3, image_mesh_min_mesh_number=2
+            image_mesh_min_mesh_pixels_per_pixel=3,
+            image_mesh_min_mesh_number=2,
         )
 
 
@@ -76,7 +79,9 @@ def test__check_adapt_background_pixels(mask, mesh_grid, image_mesh):
     )
 
     image_mesh.check_adapt_background_pixels(
-        mask=mask, mesh_grid=mesh_grid, adapt_data=adapt_data,
+        mask=mask,
+        mesh_grid=mesh_grid,
+        adapt_data=adapt_data,
     )
 
     image_mesh.check_adapt_background_pixels(