def test__figures_1d__all_are_output(
lp_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"),
)
light_profile_plotter = aplt.LightProfilePlotter(
light_profile=lp_0,
grid=sub_grid_2d_7x7,
include_1d=include_1d_all,
mat_plot_1d=mat_plot_1d,
)
light_profile_plotter.figures_1d(image=True)
assert path.join(plot_path, "image_1d.png") in plot_patch.paths
def test__visuals_with_include_2d(lp_0, grid_2d_7x7):
visuals_2d = aplt.Visuals2D(vector_field=2)
include = aplt.Include2D(
origin=True,
mask=True,
border=True,
light_profile_centres=True,
critical_curves=True,
caustics=True,
)
light_profile_plotter = aplt.LightProfilePlotter(
light_profile=lp_0, grid=grid_2d_7x7, visuals_2d=visuals_2d, include_2d=include
)
assert light_profile_plotter.visuals_with_include_2d.origin.in_list == [(0.0, 0.0)]
assert (
light_profile_plotter.visuals_with_include_2d.mask == grid_2d_7x7.mask
).all()
assert (
light_profile_plotter.visuals_with_include_2d.border
== grid_2d_7x7.mask.border_grid_sub_1.binned
).all()
assert (
light_profile_plotter.visuals_with_include_2d.light_profile_centres.in_list
== [lp_0.centre]
)
assert light_profile_plotter.visuals_with_include_2d.vector_field == 2
include = aplt.Include2D(origin=False, mask=False, border=False)
light_profile_plotter = aplt.LightProfilePlotter(
light_profile=lp_0, grid=grid_2d_7x7, visuals_2d=visuals_2d, include_2d=include
)
assert light_profile_plotter.visuals_with_include_2d.origin == None
assert light_profile_plotter.visuals_with_include_2d.mask == None
assert light_profile_plotter.visuals_with_include_2d.border == None
assert light_profile_plotter.visuals_with_include_2d.vector_field == 2
def test__figures_2d__all_are_output(
lp_0,
sub_grid_2d_7x7,
grid_2d_irregular_7x7_list,
include_2d_all,
plot_path,
plot_patch,
):
light_profile_plotter = aplt.LightProfilePlotter(
light_profile=lp_0,
grid=sub_grid_2d_7x7,
include_2d=include_2d_all,
mat_plot_2d=aplt.MatPlot2D(output=aplt.Output(plot_path, format="png")),
)
light_profile_plotter.figures_2d(image=True)
assert path.join(plot_path, "image_2d.png") in plot_patch.paths