close_figures defaults to closing all figures

Author: betaBison

gnss_lib_py/utils/visualizations.py

@@ -521,17 +521,22 @@ def plot_map(*args, sections=0, save=False, prefix="",
                      width=width, height=height)
     return figs
 
-def close_figures(figs):
+def close_figures(figs=None):
     """Closes figures.
 
+    If figs is None, then will attempt to close all matplotlib figures
+    with plt.close('all')
+
     Parameters
     ----------
-    figs : list or matplotlib.pyplot.figure
+    figs : list or matplotlib.pyplot.figure or None
         List of figures or single matplotlib figure object.
 
     """
 
-    if isinstance(figs,plt.Figure):
+    if figs is None:
+        plt.close('all')
+    elif isinstance(figs,plt.Figure):
         plt.close(figs)
     elif isinstance(figs, list):
         for fig in figs:

tests/utils/test_visualizations.py

@@ -309,7 +309,7 @@ def test_plot_metrics_by_constellation(derived):
                 for prefix in ["","test"]:
                     fig = viz.plot_metric_by_constellation(derived, row,
                                                prefix=prefix,save=False)
-                    viz.close_figures(fig)
+                    viz.close_figures()
         else:
             # string rows should cause a KeyError
             with pytest.raises(KeyError) as excinfo: