black

test_autolens/imaging/test_fit_imaging.py

@@ -299,7 +299,7 @@ def test__galaxy_model_image_dict(masked_imaging_7x7):
         galaxies=[g0, g0_linear, g2, galaxy_pix_0, galaxy_pix_1]
     )
 
-    masked_imaging_7x7.image[0] = 3.0
+    masked_imaging_7x7.data[0] = 3.0
 
     fit = al.FitImaging(dataset=masked_imaging_7x7, tracer=tracer, settings_inversion=al.SettingsInversion(use_w_tilde=False))
 

test_autolens/imaging/test_imaging.py

@@ -34,8 +34,8 @@ def test__via_tracer_from__same_as_tracer_image(self):
         )
 
         assert dataset.shape_native == (20, 20)
-        assert dataset.data.native[0, 0] != imaging_via_image.image.native[0, 0]
-        assert dataset.data.native[10, 10] == imaging_via_image.image.native[10, 10]
+        assert dataset.data.native[0, 0] != imaging_via_image.data.native[0, 0]
+        assert dataset.data.native[10, 10] == imaging_via_image.data.native[10, 10]
         assert (dataset.psf == imaging_via_image.psf).all()
         assert (dataset.noise_map == imaging_via_image.noise_map).all()
 
@@ -70,7 +70,7 @@ def test__via_deflections_and_galaxies_from__same_as_calculation_using_tracer(se
         )
 
         assert dataset.shape_native == (20, 20)
-        assert (dataset.data.native == imaging_via_image.image.native).all()
+        assert (dataset.data.native == imaging_via_image.data.native).all()
         assert (dataset.psf == imaging_via_image.psf).all()
         assert (dataset.noise_map == imaging_via_image.noise_map).all()
 
@@ -117,6 +117,6 @@ def test__simulate_imaging_from_lens__source_galaxy__compare_to_imaging(self):
         )
 
         assert dataset.shape_native == (11, 11)
-        assert (dataset.data.native == imaging_via_image.image.native).all()
+        assert (dataset.data.native == imaging_via_image.data.native).all()
         assert (dataset.psf == imaging_via_image.psf).all()
         assert (dataset.noise_map == imaging_via_image.noise_map).all()

test_autolens/lens/test_to_inversion.py

@@ -445,7 +445,7 @@ def test__inversion_imaging_from(sub_grid_2d_7x7, masked_imaging_7x7):
     tracer_to_inversion = al.TracerToInversion(
         tracer=tracer,
         dataset=masked_imaging_7x7,
-        data=masked_imaging_7x7.image,
+        data=masked_imaging_7x7.data,
         noise_map=masked_imaging_7x7.noise_map,
         w_tilde=masked_imaging_7x7.w_tilde,
         settings_pixelization=al.SettingsPixelization(use_border=False),
@@ -468,7 +468,7 @@ def test__inversion_imaging_from(sub_grid_2d_7x7, masked_imaging_7x7):
     tracer_to_inversion = al.TracerToInversion(
         tracer=tracer,
         dataset=masked_imaging_7x7,
-        data=masked_imaging_7x7.image,
+        data=masked_imaging_7x7.data,
         noise_map=masked_imaging_7x7.noise_map,
         w_tilde=masked_imaging_7x7.w_tilde,
         settings_pixelization=al.SettingsPixelization(use_border=False),
@@ -478,7 +478,7 @@ def test__inversion_imaging_from(sub_grid_2d_7x7, masked_imaging_7x7):
     inversion = tracer_to_inversion.inversion
 
     assert inversion.mapped_reconstructed_image == pytest.approx(
-        masked_imaging_7x7.image, 1.0e-2
+        masked_imaging_7x7.data, 1.0e-2
     )