def test__figures_1d__all_are_output(
mp_0,
sub_grid_2d_7x7,
grid_2d_irregular_7x7_list,
include_1d_all,
plot_path,
plot_patch,
):
mat_plot_1d = aplt.MatPlot1D(
half_light_radius_axvline=aplt.HalfLightRadiusAXVLine(color="r"),
einstein_radius_axvline=aplt.EinsteinRadiusAXVLine(color="r"),
output=aplt.Output(plot_path, format="png"),
)
mass_profile_plotter = aplt.MassProfilePlotter(
mass_profile=mp_0,
grid=sub_grid_2d_7x7,
include_1d=include_1d_all,
mat_plot_1d=mat_plot_1d,
)
mass_profile_plotter.figures_1d(convergence=True, potential=True)
assert path.join(plot_path, "convergence_1d.png") in plot_patch.paths
assert path.join(plot_path, "potential_1d.png") in plot_patch.paths
def test__figures_2d__all_are_output(
mp_0,
sub_grid_2d_7x7,
grid_2d_irregular_7x7_list,
include_2d_all,
plot_path,
plot_patch,
):
mass_profile_plotter = aplt.MassProfilePlotter(
mass_profile=mp_0,
grid=sub_grid_2d_7x7,
include_2d=include_2d_all,
mat_plot_2d=aplt.MatPlot2D(
output=aplt.Output(plot_path, format="png")),
)
mass_profile_plotter.figures_2d(
convergence=True,
potential=True,
deflections_y=True,
deflections_x=True,
magnification=True,
)
assert path.join(plot_path, "convergence_2d.png") in plot_patch.paths
assert path.join(plot_path, "potential_2d.png") in plot_patch.paths
assert path.join(plot_path, "deflections_y_2d.png") in plot_patch.paths
assert path.join(plot_path, "deflections_x_2d.png") in plot_patch.paths
assert path.join(plot_path, "magnification_2d.png") in plot_patch.paths
def test__visuals_with_include_2d(mp_0, grid_2d_7x7):
visuals_2d = aplt.Visuals2D(vector_field=2)
include = aplt.Include2D(
origin=True,
mask=True,
border=True,
mass_profile_centres=True,
critical_curves=True,
caustics=True,
)
mass_profile_plotter = aplt.MassProfilePlotter(mass_profile=mp_0,
grid=grid_2d_7x7,
visuals_2d=visuals_2d,
include_2d=include)
assert mass_profile_plotter.visuals_with_include_2d.origin.in_list == [
(0.0, 0.0)
]
assert (mass_profile_plotter.visuals_with_include_2d.mask ==
grid_2d_7x7.mask).all()
assert (mass_profile_plotter.visuals_with_include_2d.border ==
grid_2d_7x7.mask.border_grid_sub_1.binned).all()
assert (mass_profile_plotter.visuals_with_include_2d.mass_profile_centres.
in_list == [mp_0.centre])
assert mass_profile_plotter.visuals_with_include_2d.vector_field == 2
include = aplt.Include2D(origin=False, mask=False, border=False)
mass_profile_plotter = aplt.MassProfilePlotter(mass_profile=mp_0,
grid=grid_2d_7x7,
visuals_2d=visuals_2d,
include_2d=include)
assert mass_profile_plotter.visuals_with_include_2d.origin == None
assert mass_profile_plotter.visuals_with_include_2d.mask == None
assert mass_profile_plotter.visuals_with_include_2d.border == None
assert mass_profile_plotter.visuals_with_include_2d.vector_field == 2