add tests to webagg and tk backend

matplotlib · Feb 11, 2024 · f8a3028 · f8a3028
1 parent c58a062
commit f8a3028
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/lib/matplotlib/tests/test_backend_tk.py b/lib/matplotlib/tests/test_backend_tk.py
@@ -196,6 +196,19 @@ class Toolbar(NavigationToolbar2Tk):
     print("success")
 
 
+@_isolated_tk_test(success_count=1)
+def test_save_figure_return():
+    import matplotlib.pyplot as plt
+    from unittest import mock
+    fig = plt.figure()
+    prop = "tkinter.filedialog.asksaveasfilename"
+    with mock.patch(prop, return_value=("foobar.png", None)):
+        fname = fig.canvas.manager.toolbar.save_figure()
+        os.remove("foobar.png")
+        assert fname == "foobar.png"
+        print("success")
+
+
 @_isolated_tk_test(success_count=1)
 def test_canvas_focus():
     import tkinter as tk