TST: Use a temporary directory for test_save_figure_return

QuLogic · QuLogic · commit 628a68f122d5 · 2025-09-04T16:26:41.000-04:00
This avoids having to manually clean up the resulting file, which seems
flaky.
diff --git a/lib/matplotlib/tests/test_backend_qt.py b/lib/matplotlib/tests/test_backend_qt.py
@@ -215,14 +215,15 @@ def test_figureoptions():
 
 
 @pytest.mark.backend('QtAgg', skip_on_importerror=True)
-def test_save_figure_return():
+def test_save_figure_return(tmp_path):
     fig, ax = plt.subplots()
     ax.imshow([[1]])
+    expected = tmp_path / "foobar.png"
     prop = "matplotlib.backends.qt_compat.QtWidgets.QFileDialog.getSaveFileName"
-    with mock.patch(prop, return_value=("foobar.png", None)):
+    with mock.patch(prop, return_value=(str(expected), None)):
         fname = fig.canvas.manager.toolbar.save_figure()
-        os.remove("foobar.png")
-        assert fname == "foobar.png"
+        assert fname == str(expected)
+        assert expected.exists()
     with mock.patch(prop, return_value=(None, None)):
         fname = fig.canvas.manager.toolbar.save_figure()
         assert fname is None
