From 889a6997d0c632a3e0a7a6da9ad9eee4a47d3099 Mon Sep 17 00:00:00 2001 From: James Nightingale Date: Wed, 5 Jul 2023 15:11:48 +0100 Subject: [PATCH] black --- test_autolens/imaging/test_fit_imaging.py | 2 +- test_autolens/imaging/test_imaging.py | 8 ++++---- test_autolens/lens/test_to_inversion.py | 6 +++--- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/test_autolens/imaging/test_fit_imaging.py b/test_autolens/imaging/test_fit_imaging.py index c62b72fe5..ed2f15b1e 100644 --- a/test_autolens/imaging/test_fit_imaging.py +++ b/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)) diff --git a/test_autolens/imaging/test_imaging.py b/test_autolens/imaging/test_imaging.py index 7f1720ef3..320b214f9 100644 --- a/test_autolens/imaging/test_imaging.py +++ b/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() diff --git a/test_autolens/lens/test_to_inversion.py b/test_autolens/lens/test_to_inversion.py index f6912cf60..fd70f99ac 100644 --- a/test_autolens/lens/test_to_inversion.py +++ b/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 )