test: cover quick_update=True routing in the four model plotters (#680)

Author: Jammy2211

test_autolens/imaging/model/test_plotter_imaging.py

@@ -41,6 +41,22 @@ def test__fit_imaging(
 
     assert image.shape == (5, 5)
 
+def test__fit_imaging__quick_update__writes_normal_fit_subplot_only(
+    fit_imaging_x2_plane_inversion_7x7, plot_path, plot_patch
+):
+    if plot_path.exists():
+        shutil.rmtree(plot_path)
+
+    plotter = PlotterImaging(image_path=plot_path)
+
+    plotter.fit_imaging(fit=fit_imaging_x2_plane_inversion_7x7, quick_update=True)
+
+    assert str(plot_path / "fit.png") in plot_patch.paths
+    assert str(plot_path / "fit_quick.png") not in plot_patch.paths
+    assert str(plot_path / "tracer.png") not in plot_patch.paths
+    assert str(plot_path / "fit_log10.png") not in plot_patch.paths
+
+
 def test__fit_imaging_combined(
     fit_imaging_x2_plane_inversion_7x7, plot_path, plot_patch
 ):

test_autolens/interferometer/model/test_plotter_interferometer.py

@@ -16,6 +16,22 @@ def make_plotter_plotter_setup():
     return directory / "files"
 
 
+def test__fit_interferometer__quick_update__writes_normal_fit_subplot_only(
+    fit_interferometer_x2_plane_7x7,
+    plot_path,
+    plot_patch,
+):
+    plotter = PlotterInterferometer(image_path=plot_path)
+
+    plotter.fit_interferometer(
+        fit=fit_interferometer_x2_plane_7x7, quick_update=True
+    )
+
+    assert str(plot_path / "fit.png") in plot_patch.paths
+    assert str(plot_path / "fit_quick.png") not in plot_patch.paths
+    assert str(plot_path / "fit_dirty_images.png") not in plot_patch.paths
+
+
 def test__fit_interferometer(
     fit_interferometer_x2_plane_7x7,
     plot_path,

test_autolens/point/model/test_plotter_point.py

@@ -21,3 +21,17 @@ def test__fit_point(fit_point_dataset_x2_plane, plot_path, plot_patch):
     plotter.fit_point(fit=fit_point_dataset_x2_plane)
 
     assert str(plot_path / "fit.png") in plot_patch.paths
+
+
+def test__fit_point__quick_update__writes_normal_fit_subplot(
+    fit_point_dataset_x2_plane, plot_path, plot_patch
+):
+    if plot_path.exists():
+        shutil.rmtree(plot_path)
+
+    plotter = PlotterPoint(image_path=plot_path)
+
+    plotter.fit_point(fit=fit_point_dataset_x2_plane, quick_update=True)
+
+    assert str(plot_path / "fit.png") in plot_patch.paths
+    assert str(plot_path / "fit_quick.png") not in plot_patch.paths

test_autolens/weak/model/__init__.py

@@ -0,0 +1,53 @@
+from pathlib import Path
+
+import autoarray as aa
+import autolens as al
+
+import pytest
+
+from autolens.weak.model.plotter import PlotterWeak
+
+directory = Path(__file__).resolve().parent
+
+
+def _isothermal_tracer(einstein_radius=1.6, ell_comps=(0.0, 0.05)):
+    lens = al.Galaxy(
+        redshift=0.5,
+        mass=al.mp.Isothermal(
+            centre=(0.0, 0.0),
+            ell_comps=ell_comps,
+            einstein_radius=einstein_radius,
+        ),
+    )
+    source = al.Galaxy(redshift=1.0)
+    return al.Tracer(galaxies=[lens, source])
+
+
+@pytest.fixture(name="fit_weak")
+def make_fit_weak():
+    grid = aa.Grid2DIrregular(
+        values=[(0.7, 0.5), (1.0, 1.0), (-0.3, 0.6), (-1.1, -0.8)]
+    )
+    truth = _isothermal_tracer(einstein_radius=1.6)
+    dataset = al.SimulatorShearYX(noise_sigma=0.0, seed=0).via_tracer_from(
+        tracer=truth, grid=grid, name="test"
+    )
+    dataset.noise_map = aa.ArrayIrregular(values=[0.3, 0.3, 0.3, 0.3])
+    model = _isothermal_tracer(einstein_radius=1.5)
+    return al.FitWeak(dataset=dataset, tracer=model)
+
+
+@pytest.fixture(name="plot_path")
+def make_plot_path():
+    return directory / "files"
+
+
+def test__fit_weak__quick_update__writes_normal_fit_subplot(
+    fit_weak, plot_path, plot_patch
+):
+    plotter = PlotterWeak(image_path=plot_path)
+
+    plotter.fit_weak(fit=fit_weak, quick_update=True)
+
+    assert str(plot_path / "subplot_fit_weak.png") in plot_patch.paths
+    assert str(plot_path / "fit_quick.png") not in plot_patch.paths