Check if the file exists, if it exists then delete it, to avoid doubl…

…e deletion
pybamm-team · Jul 2, 2021 · 05f9cca · 05f9cca
1 parent f990ff1
commit 05f9cca
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/test_random_plot_generator.py b/test/test_random_plot_generator.py
@@ -8,8 +8,10 @@
 class TestRandomPlotGenerator(unittest.TestCase):
 
     def tearDown(self):
-        os.remove("plot.png")
-        os.remove("plot.gif")
+        if os.path.exists("plot.png"):
+            os.remove("plot.png")
+        elif os.path.exists("plot.gif"):
+            os.remove("plot.gif")
 
     def test_random_plot_generator(self):