TST: macosx savefig respects rcParam setting

greglucas · greglucas · commit 9ed0a309b953 · 2022-03-08T07:18:54.000-07:00
diff --git a/lib/matplotlib/tests/test_backend_macosx.py b/lib/matplotlib/tests/test_backend_macosx.py
@@ -1,10 +1,14 @@
+import os
+import tempfile
+
 import pytest
 
+import matplotlib as mpl
 import matplotlib.pyplot as plt
-
-
-pytest.importorskip("matplotlib.backends.backend_macosx",
-                    reason="These are mac only tests")
+try:
+    from matplotlib.backends import _macosx
+except ImportError:
+    pytest.skip("These are mac only tests", allow_module_level=True)
 
 
 @pytest.mark.backend('macosx')
@@ -18,3 +22,26 @@ 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(monkeypatch):
+
+    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}"
+
+    monkeypatch.setattr(_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"
