def test__visualizer_with_preloaded_critical_curves_and_caustics_is_setup(
self,
masked_imaging_7x7,
tracer_x2_plane_7x7,
include_all,
plot_path,
plot_patch,
):
visualizer = vis.PhaseDatasetVisualizer(
masked_dataset=masked_imaging_7x7, image_path=plot_path)
assert visualizer.include.preloaded_critical_curves == None
assert visualizer.include.preloaded_caustics == None
visualizer = visualizer.new_visualizer_with_preloaded_critical_curves_and_caustics(
preloaded_critical_curves=1, preloaded_caustics=2)
assert visualizer.include.preloaded_critical_curves == 1
assert visualizer.include.preloaded_caustics == 2
visualizer.include.critical_curves = False
visualizer.include.caustics = False
visualizer.visualize_ray_tracing(tracer=tracer_x2_plane_7x7,
during_analysis=True)
assert plot_path + "subplots/subplot_tracer.png" in plot_patch.paths
def test__visualize_stochastic_histogram(self, masked_imaging_7x7,
plot_path, plot_patch):
visualizer = vis.PhaseDatasetVisualizer(
masked_dataset=masked_imaging_7x7)
visualizer.visualize_stochastic_histogram(
paths=af.Paths(),
log_evidences=[1.0, 2.0, 1.0, 2.0, 3.0, 2.5],
max_log_evidence=3.0,
)
assert (path.join(plot_path, "image", "other",
"stochastic_histogram.png") in plot_patch.paths)
def test__visualizes_ray_tracing_using_configs(
self,
masked_imaging_7x7,
tracer_x2_plane_7x7,
include_all,
plot_path,
plot_patch,
):
if os.path.exists(plot_path):
shutil.rmtree(plot_path)
visualizer = vis.PhaseDatasetVisualizer(
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_ray_tracing(paths=af.Paths(),
tracer=tracer_x2_plane_7x7,
during_analysis=False)
assert (path.join(plot_path, "image", "ray_tracing",
"subplot_tracer.png") in plot_patch.paths)
assert (path.join(plot_path, "image", "ray_tracing", "image.png")
in plot_patch.paths)
assert (path.join(plot_path, "image", "ray_tracing",
"source_plane.png") in plot_patch.paths)
assert (path.join(plot_path, "image", "ray_tracing", "convergence.png")
in plot_patch.paths)
assert (path.join(plot_path, "image", "ray_tracing", "potential.png")
not in plot_patch.paths)
assert (path.join(plot_path, "image", "ray_tracing",
"deflections_y.png") not in plot_patch.paths)
assert (path.join(plot_path, "image", "ray_tracing",
"deflections_x.png") not in plot_patch.paths)
assert (path.join(plot_path, "image", "ray_tracing",
"magnification.png") in plot_patch.paths)
convergence = al.util.array.numpy_array_2d_from_fits(
file_path=path.join(plot_path, "image", "ray_tracing", "fits",
"convergence.fits"),
hdu=0,
)
assert convergence.shape == (5, 5)
def test__visualize_stochastic_histogram(self, masked_imaging_7x7,
plot_path, plot_patch):
visualizer = vis.PhaseDatasetVisualizer(
masked_dataset=masked_imaging_7x7, image_path=plot_path)
visualizer.visualize_stochastic_histogram(
log_evidences=[1.0, 2.0, 1.0, 2.0, 3.0, 2.5],
max_log_evidence=3.0,
during_analysis=True,
)
assert plot_path + "other/stochastic_histogram.png" not in plot_patch.paths
visualizer.visualize_stochastic_histogram(
log_evidences=[1.0, 2.0, 1.0, 2.0, 3.0, 2.5],
max_log_evidence=3.0,
during_analysis=False,
)
assert plot_path + "other/stochastic_histogram.png" in plot_patch.paths
def test__visualizes_ray_tracing_using_configs(
self,
masked_imaging_7x7,
tracer_x2_plane_7x7,
include_all,
plot_path,
plot_patch,
):
if os.path.exists(plot_path):
shutil.rmtree(plot_path)
visualizer = vis.PhaseDatasetVisualizer(
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_ray_tracing(
tracer=tracer_x2_plane_7x7, during_analysis=False
)
assert plot_path + "subplots/subplot_tracer.png" in plot_patch.paths
assert plot_path + "ray_tracing/profile_image.png" in plot_patch.paths
assert plot_path + "ray_tracing/source_plane.png" in plot_patch.paths
assert plot_path + "ray_tracing/convergence.png" in plot_patch.paths
assert plot_path + "ray_tracing/potential.png" not in plot_patch.paths
assert plot_path + "ray_tracing/deflections_y.png" not in plot_patch.paths
assert plot_path + "ray_tracing/deflections_x.png" not in plot_patch.paths
assert plot_path + "ray_tracing/magnification.png" in plot_patch.paths
convergence = al.util.array.numpy_array_2d_from_fits(
file_path=plot_path + "ray_tracing/fits/convergence.fits", hdu=0
)
assert convergence.shape == (5, 5)