Moved mock_inversion to autoarray from automodel

Author: Jammy2211

test_autoarray/mock/mock_inversion.py

@@ -0,0 +1,88 @@
+import numpy as np
+
+
+class MockGeometry(object):
+    def __init__(
+        self,
+        pixel_centres=None,
+        pixel_neighbors=np.array([1]),
+        pixel_neighbors_size=np.array([1]),
+    ):
+
+        self.pixel_scales = (1.0, 1.0)
+        self.origin = (0.0, 0.0)
+        self.pixel_centres = pixel_centres
+
+        self.pixel_neighbors = pixel_neighbors.astype("int")
+        self.pixel_neighbors_size = pixel_neighbors_size.astype("int")
+
+
+class MockPixelization(object):
+    def __init__(self, value, grid=None):
+        self.value = value
+        self.grid = grid
+
+    # noinspection PyUnusedLocal,PyShadowingNames
+    def mapper_from_grid_and_pixelization_grid(
+        self, grid, pixelization_grid, inversion_uses_border, hyper_image=None
+    ):
+        return self.value
+
+    def pixelization_grid_from_grid(self, grid, cluster_grid, hyper_image):
+        if hyper_image is None:
+            return self.grid
+        else:
+            return self.grid * hyper_image
+
+
+class MockRegularization(object):
+    def __init__(self, matrix_shape):
+        self.shape = matrix_shape
+
+    def regularization_matrix_from_pixel_neighbors(
+        self, pixel_neighbors, pixel_neighbors_size
+    ):
+        return np.array([[1.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, 1.0]])
+
+    def regularization_matrix_from_mapper(self, mapper):
+        return np.array([[1.0, 0.0, 0.0], [0.0, 1.0, 0.0], [0.0, 0.0, 1.0]])
+
+
+class MockRegMapper(object):
+
+    def __init__(self, pixel_neighbors=None, pixel_neighbors_size=None, pixel_signals=None):
+        self.pixel_neighbors = pixel_neighbors
+        self.pixel_neighbors_size = pixel_neighbors_size
+        self.pixel_signals = pixel_signals
+
+    def pixel_signals_from_signal_scale(self, signal_scale):
+        return self.pixel_signals
+
+
+class MockMapper(object):
+    def __init__(self, matrix_shape, grid=None):
+
+        self.grid = grid
+        self.mapping_matrix = np.ones(matrix_shape)
+        self.geometry = MockGeometry()
+
+
+class MockConvolver(object):
+    def __init__(self, matrix_shape):
+        self.shape = matrix_shape
+
+    def convolve_mapping_matrix(self, mapping_matrix):
+        return np.ones(self.shape)
+
+
+class MockInversion(object):
+    def __init__(self):
+        self.blurred_mapping_matrix = np.zeros((1, 1))
+        self.regularization_matrix = np.zeros((1, 1))
+        self.curvature_matrix = np.zeros((1, 1))
+        self.curvature_reg_matrix = np.zeros((1, 1))
+        self.solution_vector = np.zeros((1))
+
+    @property
+    def reconstructed_image(self):
+        return np.zeros((1, 1))

test_autoarray/unit/operators/inversion/test_inversions.py

@@ -3,7 +3,7 @@
 import numpy as np
 import pytest
 
-from test_automodel.mock import mock_inversion
+from test_autoarray.mock import mock_inversion
 
 
 class TestRegularizationTerm:

test_autoarray/unit/operators/inversion/test_mappers.py

@@ -3,7 +3,7 @@
 
 import autoarray as aa
 from test_autoarray.mock.mock_grids import MockPixelizationGrid
-from test_automodel.mock.mock_inversion import MockGeometry
+from test_autoarray.mock.mock_inversion import MockGeometry
 
 
 def grid_to_pixel_pixels_via_nearest_neighbour(grid, pixel_centers):

test_autoarray/unit/operators/inversion/test_regularization.py

@@ -1,7 +1,7 @@
 import autoarray as aa
 import numpy as np
 
-from test_automodel.mock.mock_inversion import MockRegMapper
+from test_autoarray.mock.mock_inversion import MockRegMapper
 
 class TestRegularizationConstant:
     def test__regularization_matrix__compare_to_regularization_util(self):