TST: macosx savefig respects rcParam setting

greglucas · greglucas · commit 647451d6f9b5 · 2022-01-10T08:56:23.000-07:00
diff --git a/lib/matplotlib/tests/test_backend_macosx.py b/lib/matplotlib/tests/test_backend_macosx.py
@@ -1,5 +1,10 @@
+import os
+import tempfile
+
 import pytest
 
+import matplotlib as mpl
+from matplotlib.backends import _macosx
 import matplotlib.pyplot as plt
 
 
@@ -18,3 +23,32 @@ def test_cached_renderer():
     fig = plt.figure(2)
     fig.draw_without_rendering()
     assert fig._cachedRenderer is not None
+
+
+@pytest.mark.backend('macosx')
+def test_savefig_rcparam():
+    # Store the save file so we can overwrite it and
+    # then put back the real one after this
+    old_save_file_func = _macosx.choose_save_file
+
+    def new_choose_save_file(title, directory, filename):
+        # Replacement function instead of opening a GUI window
+        # Make a new directory for testing the update of the rcParams
+        os.makedirs(f"{directory}/test")
+        return f"{directory}/test/{filename}"
+
+    _macosx.choose_save_file = new_choose_save_file
+    fig = plt.figure()
+    with tempfile.TemporaryDirectory() as temp_dir, \
+            mpl.rc_context({"savefig.directory": temp_dir}):
+        fig.canvas.toolbar.save_figure()
+        # Check the saved location got created
+        save_file = f"{temp_dir}/test/{fig.canvas.get_default_filename()}"
+        assert os.path.exists(save_file)
+
+        # Check the savefig.directory rcParam got updated because
+        # we added a subdirectory "test"
+        assert mpl.rcParams["savefig.directory"] == f"{temp_dir}/test"
+
+    # Replace the original function
+    _macosx.choose_save_file = old_save_file_func
