unit test fix

Author: Jammy2211

autoarray/inversion/inversion/imaging/w_tilde.py

@@ -523,9 +523,17 @@ def mapped_reconstructed_data_dict(self) -> Dict[LinearObj, Array2D]:
                     reconstruction=reconstruction,
                 )
 
-                mapped_reconstructed_image = self.psf.convolve_image_no_blurring(
-                    image=mapped_reconstructed_image, mask=self.mask
-                ).array
+                # mapped_reconstructed_image = self.psf.convolve_image_no_blurring(
+                #     image=mapped_reconstructed_image, mask=self.mask
+                # ).array
+
+                mapped_reconstructed_image = Array2D(
+                    values=mapped_reconstructed_image, mask=self.mask
+                )
+
+                mapped_reconstructed_image = self.convolver.convolve_image_no_blurring(
+                    image=mapped_reconstructed_image
+                )
 
                 mapped_reconstructed_image = Array2D(
                     values=np.array(mapped_reconstructed_image), mask=self.mask

test_autoarray/dataset/plot/test_imaging_plotters.py

@@ -81,4 +81,4 @@ def test__output_as_fits__correct_output_format(
         file_path=path.join(plot_path, "data.fits"), hdu=0
     )
 
-    assert image_from_plot.shape == (5, 5)
+    assert image_from_plot.shape == (7, 7)

test_autoarray/inversion/inversion/test_factory.py

@@ -382,6 +382,12 @@ def test__inversion_imaging__linear_obj_func_with_w_tilde(
     assert inversion_mapping.curvature_matrix == pytest.approx(
         inversion_w_tilde.curvature_matrix, 1.0e-4
     )
+    assert inversion_mapping.curvature_reg_matrix == pytest.approx(
+        inversion_w_tilde.curvature_reg_matrix, 1.0e-4
+    )
+    assert inversion_mapping.reconstruction == pytest.approx(
+        inversion_w_tilde.reconstruction, 1.0e-4
+    )
     assert inversion_mapping.mapped_reconstructed_image == pytest.approx(
         inversion_w_tilde.mapped_reconstructed_image, 1.0e-4
     )