fix mapped valued tests by converting to JAX

Author: Jammy2211

autoarray/inversion/inversion/mapper_valued.py

@@ -57,7 +57,7 @@ def values_masked(self):
         values = self.values
 
         if self.mesh_pixel_mask is not None:
-            values[self.mesh_pixel_mask] = 0.0
+            values = values.at[self.mesh_pixel_mask].set(0.0)
 
         return values
 
@@ -187,7 +187,7 @@ def mapped_reconstructed_image_from(
         mapping_matrix = self.mapper.mapping_matrix
 
         if self.mesh_pixel_mask is not None:
-            mapping_matrix[:, self.mesh_pixel_mask] = 0.0
+            mapping_matrix = mapping_matrix.at[:, self.mesh_pixel_mask].set(0.0)
 
         return Array2D(
             values=inversion_util.mapped_reconstructed_data_via_mapping_matrix_from(

autoarray/inversion/pixelization/mappers/abstract.py

@@ -282,6 +282,7 @@ def pixel_signals_from(self, signal_scale: float) -> np.ndarray:
             A factor which controls how rapidly the smoothness of regularization varies from high signal regions to
             low signal regions.
         """
+
         return mapper_util.adaptive_pixel_signals_from(
             pixels=self.pixels,
             signal_scale=signal_scale,

test_autoarray/inversion/inversion/test_mapper_valued.py

@@ -1,3 +1,4 @@
+import jax.numpy as jnp
 import numpy as np
 import pytest
 
@@ -145,7 +146,7 @@ def test__magnification_via_mesh_from__with_pixel_mask():
         pixel_scales=(0.5, 0.5),
     )
 
-    magnification = [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
+    magnification = jnp.array([1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0])
 
     source_plane_mesh_grid = aa.Mesh2DVoronoi(
         values=np.array(
@@ -168,15 +169,15 @@ def test__magnification_via_mesh_from__with_pixel_mask():
         parameters=3,
         source_plane_mesh_grid=source_plane_mesh_grid,
         mask=mask,
-        mapping_matrix=np.ones((12, 10)),
+        mapping_matrix=jnp.ones((12, 10)),
     )
 
     mesh_pixel_mask = np.array(
         [True, True, True, True, True, True, True, True, False, False]
     )
 
     mapper_valued = aa.MapperValued(
-        values=np.array(magnification), mapper=mapper, mesh_pixel_mask=mesh_pixel_mask
+        values=magnification, mapper=mapper, mesh_pixel_mask=mesh_pixel_mask
     )
 
     magnification = mapper_valued.magnification_via_mesh_from()
@@ -199,7 +200,7 @@ def test__magnification_via_interpolation_from():
         parameters=4,
         mask=mask,
         interpolated_array=magnification,
-        mapping_matrix=np.ones((4, 4)),
+        mapping_matrix=jnp.ones((4, 4)),
     )
 
     mapper_valued = aa.MapperValued(values=np.array(magnification), mapper=mapper)