Improve test cleanup for monte_carlo_plots_all_save test

Co-authored-by: Gui-FernandesBR <63590233+Gui-FernandesBR@users.noreply.github.com>

Author: Copilot

tests/integration/simulation/test_monte_carlo.py

@@ -11,14 +11,18 @@
 
 def _post_test_file_cleanup():
     """Clean monte carlo files after test session if they exist."""
-    if os.path.exists("monte_carlo_class_example.kml"):
-        os.remove("monte_carlo_class_example.kml")
-    if os.path.exists("monte_carlo_test.errors.txt"):
-        os.remove("monte_carlo_test.errors.txt")
-    if os.path.exists("monte_carlo_test.inputs.txt"):
-        os.remove("monte_carlo_test.inputs.txt")
-    if os.path.exists("monte_carlo_test.outputs.txt"):
-        os.remove("monte_carlo_test.outputs.txt")
+    files_to_cleanup = [
+        "monte_carlo_class_example.kml",
+        "monte_carlo_test.errors.txt",
+        "monte_carlo_test.inputs.txt",
+        "monte_carlo_test.outputs.txt",
+        "test_histogram_apogee.png",
+        "test_multi_apogee.png",
+        "test_multi_x_impact.png",
+    ]
+    for filepath in files_to_cleanup:
+        if os.path.exists(filepath):
+            os.remove(filepath)
 
 
 @pytest.mark.slow
@@ -148,16 +152,13 @@ def test_monte_carlo_plots_all_save(monte_carlo_calisto_pre_loaded):
             keys="apogee", filename="test_histogram.png"
         )
         assert os.path.exists("test_histogram_apogee.png")
-        os.remove("test_histogram_apogee.png")
 
         # Test saving with multiple keys
         monte_carlo_calisto_pre_loaded.plots.all(
             keys=["apogee", "x_impact"], filename="test_multi.png"
         )
         assert os.path.exists("test_multi_apogee.png")
         assert os.path.exists("test_multi_x_impact.png")
-        os.remove("test_multi_apogee.png")
-        os.remove("test_multi_x_impact.png")
     finally:
         _post_test_file_cleanup()