Small typo in test adaptation.

Author: freddieknets

tests/data/all_tests.list

@@ -159,14 +159,22 @@ tests/test_line_tools.py::test_line_accessor[1-ContextCpu]
 tests/test_line_tools.py::test_line_accessor[1-ContextCpu:auto]
 tests/test_line_tools.py::test_line_accessor[2-ContextCpu]
 tests/test_line_tools.py::test_line_accessor[2-ContextCpu:auto]
-tests/test_lossmap.py::test_lossmap_everest[B1H-ContextCpu]
-tests/test_lossmap.py::test_lossmap_everest[B1H-ContextCpu:auto]
-tests/test_lossmap.py::test_lossmap_everest[B2V-ContextCpu]
-tests/test_lossmap.py::test_lossmap_everest[B2V-ContextCpu:auto]
-tests/test_lossmap.py::test_lossmap_everest[B1V_crystals-ContextCpu]
-tests/test_lossmap.py::test_lossmap_everest[B1V_crystals-ContextCpu:auto]
-tests/test_lossmap.py::test_lossmap_everest[B2H_crystals-ContextCpu]
-tests/test_lossmap.py::test_lossmap_everest[B2H_crystals-ContextCpu:auto]
+tests/test_lossmap.py::test_lossmap_everest_with_plot[B1H-ContextCpu]
+tests/test_lossmap.py::test_lossmap_everest_with_plot[B1H-ContextCpu:auto]
+tests/test_lossmap.py::test_lossmap_everest_with_plot[B2V-ContextCpu]
+tests/test_lossmap.py::test_lossmap_everest_with_plot[B2V-ContextCpu:auto]
+tests/test_lossmap.py::test_lossmap_everest_with_plot[B1V_crystals-ContextCpu]
+tests/test_lossmap.py::test_lossmap_everest_with_plot[B1V_crystals-ContextCpu:auto]
+tests/test_lossmap.py::test_lossmap_everest_with_plot[B2H_crystals-ContextCpu]
+tests/test_lossmap.py::test_lossmap_everest_with_plot[B2H_crystals-ContextCpu:auto]
+tests/test_lossmap.py::test_lossmap_everest_without_plot[B1H-ContextCpu]
+tests/test_lossmap.py::test_lossmap_everest_without_plot[B1H-ContextCpu:auto]
+tests/test_lossmap.py::test_lossmap_everest_without_plot[B2V-ContextCpu]
+tests/test_lossmap.py::test_lossmap_everest_without_plot[B2V-ContextCpu:auto]
+tests/test_lossmap.py::test_lossmap_everest_without_plot[B1V_crystals-ContextCpu]
+tests/test_lossmap.py::test_lossmap_everest_without_plot[B1V_crystals-ContextCpu:auto]
+tests/test_lossmap.py::test_lossmap_everest_without_plot[B2H_crystals-ContextCpu]
+tests/test_lossmap.py::test_lossmap_everest_without_plot[B2H_crystals-ContextCpu:auto]
 tests/test_rf_sweep.py::test_rf_sweep[DP pos-ContextCpu]
 tests/test_rf_sweep.py::test_rf_sweep[DP pos-ContextCpu:auto]
 tests/test_rf_sweep.py::test_rf_sweep[DP neg-ContextCpu]

tests/test_lossmap.py

@@ -66,7 +66,7 @@ def _test_lossmap_everest(beam, plane, npart, interpolation, ignore_crystals, te
     line.track(part, num_turns=2)
     line.scattering.disable()
     this_id = f"B{beam}{plane}-{npart}-{interpolation}-{ignore_crystals}-{test_context}"
-    _assert_lossmap(beam, npart, line, part, tcp, interpolation, ignore_crystals, 'EverestCollimator', 'EverestCrystal', this_id)
+    ThisLM = _assert_lossmap(beam, npart, line, part, tcp, interpolation, ignore_crystals, 'EverestCollimator', 'EverestCrystal', this_id)
     if do_plot:
         ThisLM.plot(show=False, savefig=f"test-{this_id}.jpg")
         assert Path(f"test-{this_id}.jpg").exists()
@@ -142,3 +142,4 @@ def _assert_lossmap(beam, npart, line, part, tcp, interpolation, ignore_crystals
                 assert np.all([s < lm['machine_length'] for s in lm['aperture']['s']])
         assert lm['interpolation'] == interpolation
         assert lm['reversed'] == line_is_reversed
+    return ThisLM