def test__visualizes_imaging_using_configs(
self, masked_imaging_7x7, include_all, plot_path, plot_patch
):
visualizer = vis.PhaseImagingVisualizer(
masked_dataset=masked_imaging_7x7, image_path=plot_path
)
visualizer = visualizer.new_visualizer_with_preloaded_critical_curves_and_caustics(
preloaded_critical_curves=include_all.preloaded_critical_curves,
preloaded_caustics=include_all.preloaded_caustics,
)
visualizer.visualize_imaging()
assert plot_path + "subplots/subplot_imaging.png" in plot_patch.paths
assert plot_path + "imaging/image.png" in plot_patch.paths
assert plot_path + "imaging/noise_map.png" not in plot_patch.paths
assert plot_path + "imaging/psf.png" in plot_patch.paths
assert plot_path + "imaging/signal_to_noise_map.png" not in plot_patch.paths
assert (
plot_path + "imaging/absolute_signal_to_noise_map.png"
not in plot_patch.paths
)
assert plot_path + "imaging/potential_chi_squared_map.png" in plot_patch.paths
def test__visualizes_hyper_images_using_config(
self,
masked_imaging_7x7,
hyper_model_image_7x7,
include_all,
hyper_galaxy_image_path_dict_7x7,
plot_path,
plot_patch,
):
visualizer = vis.PhaseImagingVisualizer(
masked_dataset=masked_imaging_7x7, image_path=plot_path)
visualizer = visualizer.new_visualizer_with_preloaded_critical_curves_and_caustics(
preloaded_critical_curves=include_all.preloaded_critical_curves,
preloaded_caustics=include_all.preloaded_caustics,
)
visualizer.visualize_hyper_images(
hyper_galaxy_image_path_dict=hyper_galaxy_image_path_dict_7x7,
hyper_model_image=hyper_model_image_7x7,
)
assert plot_path + "hyper/hyper_model_image.png" in plot_patch.paths
assert plot_path + "hyper/subplot_hyper_galaxy_images.png" in plot_patch.paths
def __init__(
self,
masked_imaging,
settings,
cosmology,
image_path=None,
results=None,
log_likelihood_cap=None,
):
super().__init__(
settings=settings,
cosmology=cosmology,
results=results,
log_likelihood_cap=log_likelihood_cap,
)
self.visualizer = visualizer.PhaseImagingVisualizer(
masked_dataset=masked_imaging, image_path=image_path, results=results
)
self.visualizer.visualize_hyper_images(
hyper_galaxy_image_path_dict=self.hyper_galaxy_image_path_dict,
hyper_model_image=self.hyper_model_image,
)
self.masked_dataset = masked_imaging
def test__visualizes_hyper_images_using_config(
self,
masked_imaging_7x7,
hyper_model_image_7x7,
include_all,
hyper_galaxy_image_path_dict_7x7,
masked_imaging_fit_x2_plane_7x7,
plot_path,
plot_patch,
):
visualizer = vis.PhaseImagingVisualizer(
masked_dataset=masked_imaging_7x7)
visualizer = visualizer.new_visualizer_with_preloaded_critical_curves_and_caustics(
preloaded_critical_curves=include_all.preloaded_critical_curves,
preloaded_caustics=include_all.preloaded_caustics,
)
visualizer.visualize_hyper_images(
paths=af.Paths(),
hyper_galaxy_image_path_dict=hyper_galaxy_image_path_dict_7x7,
hyper_model_image=hyper_model_image_7x7,
contribution_maps_of_galaxies=masked_imaging_fit_x2_plane_7x7.
tracer.contribution_maps_of_planes,
)
assert (path.join(plot_path, "image", "hyper", "hyper_model_image.png")
in plot_patch.paths)
assert (path.join(plot_path, "image", "hyper",
"subplot_hyper_images_of_galaxies.png")
in plot_patch.paths)
assert (path.join(plot_path, "image", "hyper",
"subplot_contribution_maps_of_galaxies.png")
not in plot_patch.paths)
def test__visualizes_imaging_using_configs(self, masked_imaging_7x7,
include_all, plot_path,
plot_patch):
if os.path.exists(plot_path):
shutil.rmtree(plot_path)
visualizer = vis.PhaseImagingVisualizer(
masked_dataset=masked_imaging_7x7)
visualizer = visualizer.new_visualizer_with_preloaded_critical_curves_and_caustics(
preloaded_critical_curves=include_all.preloaded_critical_curves,
preloaded_caustics=include_all.preloaded_caustics,
)
visualizer.visualize_imaging(paths=af.Paths())
assert (path.join(plot_path, "image", "imaging", "subplot_imaging.png")
in plot_patch.paths)
assert path.join(plot_path, "image", "imaging",
"image.png") in plot_patch.paths
assert (path.join(plot_path, "image", "imaging", "noise_map.png")
not in plot_patch.paths)
assert path.join(plot_path, "image", "imaging",
"psf.png") in plot_patch.paths
assert (path.join(plot_path, "image", "imaging",
"inverse_noise_map.png") in plot_patch.paths)
assert (path.join(plot_path, "image", "imaging",
"signal_to_noise_map.png") not in plot_patch.paths)
assert (path.join(plot_path, "image", "imaging",
"absolute_signal_to_noise_map.png")
not in plot_patch.paths)
assert (path.join(plot_path, "image", "imaging",
"potential_chi_squared_map.png") in plot_patch.paths)
def __init__(self, masked_imaging, settings, cosmology, results=None):
super().__init__(
masked_dataset=masked_imaging,
settings=settings,
cosmology=cosmology,
results=results,
)
self.visualizer = visualizer.PhaseImagingVisualizer(
masked_dataset=masked_imaging)
def __init__(self,
masked_imaging,
cosmology,
image_path=None,
results=None):
super(Analysis, self).__init__(cosmology=cosmology, results=results)
self.visualizer = visualizer.PhaseImagingVisualizer(
masked_dataset=masked_imaging,
image_path=image_path,
results=results)
self.masked_dataset = masked_imaging
def __init__(self,
masked_imaging,
cosmology,
image_path=None,
results=None):
super(Analysis, self).__init__(cosmology=cosmology, results=results)
self.visualizer = visualizer.PhaseImagingVisualizer(
masked_dataset=masked_imaging,
image_path=image_path,
results=results)
self.visualizer.visualize_hyper_images(
hyper_galaxy_image_path_dict=self.hyper_galaxy_image_path_dict,
hyper_model_image=self.hyper_model_image,
)
self.masked_dataset = masked_imaging
def test__source_and_lens__visualizes_fit_and_inversion_using_configs(
self,
masked_imaging_7x7,
masked_imaging_fit_x2_plane_inversion_7x7,
include_all,
plot_path,
plot_patch,
):
if os.path.exists(plot_path):
shutil.rmtree(plot_path)
visualizer = vis.PhaseImagingVisualizer(
masked_dataset=masked_imaging_7x7, image_path=plot_path)
visualizer = visualizer.new_visualizer_with_preloaded_critical_curves_and_caustics(
preloaded_critical_curves=include_all.preloaded_critical_curves,
preloaded_caustics=include_all.preloaded_caustics,
)
visualizer.visualize_fit(fit=masked_imaging_fit_x2_plane_inversion_7x7,
during_analysis=False)
assert plot_path + "subplots/subplot_fit_imaging.png" in plot_patch.paths
assert plot_path + "fit_imaging/image.png" in plot_patch.paths
assert plot_path + "fit_imaging/noise_map.png" not in plot_patch.paths
assert plot_path + "fit_imaging/signal_to_noise_map.png" not in plot_patch.paths
assert plot_path + "fit_imaging/model_image.png" in plot_patch.paths
assert plot_path + "fit_imaging/residual_map.png" not in plot_patch.paths
assert plot_path + "fit_imaging/normalized_residual_map.png" in plot_patch.paths
assert plot_path + "fit_imaging/chi_squared_map.png" in plot_patch.paths
assert (plot_path + "fit_imaging/subtracted_image_of_plane_0.png"
in plot_patch.paths)
assert (plot_path + "fit_imaging/subtracted_image_of_plane_1.png"
in plot_patch.paths)
assert (plot_path + "fit_imaging/model_image_of_plane_0.png"
not in plot_patch.paths)
assert (plot_path + "fit_imaging/model_image_of_plane_1.png"
not in plot_patch.paths)
assert plot_path + "fit_imaging/plane_image_of_plane_0.png" in plot_patch.paths
assert plot_path + "fit_imaging/plane_image_of_plane_1.png" in plot_patch.paths
assert plot_path + "subplots/subplot_inversion.png" in plot_patch.paths
assert plot_path + "inversion/reconstructed_image.png" in plot_patch.paths
assert plot_path + "inversion/reconstruction.png" in plot_patch.paths
assert plot_path + "inversion/errors.png" not in plot_patch.paths
assert plot_path + "inversion/residual_map.png" not in plot_patch.paths
assert (plot_path + "inversion/normalized_residual_map.png"
not in plot_patch.paths)
assert plot_path + "inversion/chi_squared_map.png" in plot_patch.paths
assert (plot_path + "inversion/regularization_weight_map.png"
not in plot_patch.paths)
assert (plot_path + "inversion/interpolated_reconstruction.png"
not in plot_patch.paths)
assert plot_path + "inversion/interpolated_errors.png" in plot_patch.paths
image = al.util.array.numpy_array_2d_from_fits(
file_path=plot_path + "fit_imaging/fits/image.fits", hdu=0)
assert image.shape == (5, 5)
image = al.util.array.numpy_array_2d_from_fits(
file_path=plot_path +
"inversion/fits/interpolated_reconstruction.fits",
hdu=0,
)
assert image.shape == (7, 7)