diff --git a/pygmt/tests/test_figure.py b/pygmt/tests/test_figure.py index fa72d0926af..a01eecfd3c1 100644 --- a/pygmt/tests/test_figure.py +++ b/pygmt/tests/test_figure.py @@ -10,6 +10,7 @@ import pytest from pygmt import Figure, set_display from pygmt.exceptions import GMTInvalidInput +from pygmt.helpers import GMTTempFile def test_figure_region(): @@ -102,10 +103,9 @@ def test_figure_savefig_filename_with_spaces(): """ fig = Figure() fig.basemap(region=[0, 1, 0, 1], projection="X1c/1c", frame=True) - fname = "test_figure_savefig filename with spaces.png" - fig.savefig(fname) - assert os.path.exists(fname) - os.remove(fname) + with GMTTempFile(prefix="pygmt-filename with spaces", suffix=".png") as imgfile: + fig.savefig(imgfile.name) + assert os.path.exists(imgfile.name) def test_figure_savefig():