Free test figures after use

OGGM · Aug 20, 2024 · 8485ccc · 8485ccc
1 parent 7580f19
commit 8485ccc
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/subtests/subtest/helpers.py b/tests/subtests/subtest/helpers.py
@@ -20,7 +20,17 @@ def bar(height, **kwargs):
     return fig
 
 
+def _free_fig(func):
+    @wraps(func)
+    def free_wrapper(*args, **kwargs):
+        fig = func(*args, **kwargs)
+        plt.close(fig)
+        return fig
+    return free_wrapper
+
+
 def figure_test(test_function, **kwargs):
+    @_free_fig
     @pytest.mark.image
     @pytest.mark.hash
     @pytest.mark.mpl_image_compare(savefig_kwargs={'metadata': {'Software': None}}, **kwargs)