Merge branch 'enh/close_figures_in_tests' into 'main'

Explicitly close figures to avoid racking up memory

See merge request !75

tests/test_live_view.py

@@ -134,6 +134,8 @@ def started(spectrometer, plot_timetrace, in_process, in_gitlab_ci):
         for view in views:
             view.process.terminate()
 
+    plt.close(spectrometer.fig)
+
 
 @pytest.fixture
 def stopped(started: list[LiveViewT], in_process: bool):

tests/test_serialization.py

@@ -6,6 +6,8 @@ import time
 from tempfile import mkdtemp
 
 import pytest
+from matplotlib import pyplot as plt
+
 from python_spectrometer import Spectrometer, daq
 from qutil.plotting import is_using_mpl_gui_backend
 
@@ -55,7 +57,6 @@ def spectrometer(monkeypatch, relative_paths: bool, threaded_acquisition: bool):
                          relative_paths=relative_paths)
     speck.savepath.mkdir(parents=True, exist_ok=True)
 
-    try:
     os.chdir(speck.savepath)
 
     # XXX: need to block to avoid timing issues when running tests
@@ -71,12 +72,13 @@ def spectrometer(monkeypatch, relative_paths: bool, threaded_acquisition: bool):
         advance_frames(speck, 2)
 
     yield speck
-    finally:
+
     for file in speck.files:
         remove_file_if_exists(file)
     remove_dir_if_exists(speck.savepath)
 
     os.chdir(cwd)
+    plt.close(speck.fig)
 
 
 @pytest.fixture
@@ -84,11 +86,10 @@ def serialized(done_saving: Spectrometer):
     spectrometer = done_saving
     stem = ''.join(random.choices(string.ascii_letters, k=10))
 
-    try:
     spectrometer.serialize_to_disk(stem)
 
     yield spectrometer.savepath / stem
-    finally:
+
     exts = ['_files.txt']
     if (spectrometer.savepath / stem).is_file():
         remove_file_if_exists(spectrometer.savepath / stem)
@@ -97,6 +98,8 @@ def serialized(done_saving: Spectrometer):
     for ext in exts:
         remove_file_if_exists(spectrometer.savepath / f'{stem}{ext}')
 
+    plt.close(spectrometer.fig)
+
 
 @pytest.fixture
 def done_saving(spectrometer: Spectrometer):