modification in plot tests (plt.clf) and if/else in datasets submodule

fernandezfran · Feb 1, 2024 · 4e83af7 · 4e83af7
1 parent eaea3db
commit 4e83af7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/galpynostatic/datasets/__init__.py b/galpynostatic/datasets/__init__.py
@@ -69,5 +69,5 @@ def load_dataset(geometry="spherical"):
     """
     if geometry not in ("spherical", "cylindrical", "planar"):
         raise ValueError(f"{geometry} is not a valid geometry.")
-
-    return pd.read_csv(PATH / f"{geometry}.csv")
+    else:
+        return pd.read_csv(PATH / f"{geometry}.csv")
diff --git a/tests/test_plot.py b/tests/test_plot.py
@@ -61,6 +61,7 @@ def test_plot_versus_data(
         greg.plot.versus_data(
             experiment["C_rates"], experiment["soc"], ax=test_ax
         )
+        plt.clf()
 
         # ref plot
         ref_ax = fig_ref.subplots()
@@ -97,6 +98,7 @@ def test_plot_in_render_map(
         test_ax = fig_test.subplots()
         greg.plot.render_map(ax=test_ax)
         greg.plot.in_render_map(experiment["C_rates"], ax=test_ax)
+        plt.clf()
 
         # ref plot
         ref_ax = fig_ref.subplots()