Let Figure.savefig support filenames with spaces (GenericMappingTools…

…#1116)

Co-authored-by: Wei Ji <[email protected]>

pygmt/figure.py

@@ -190,6 +190,8 @@ def psconvert(self, **kwargs):
         # Default cropping the figure to True
         if "A" not in kwargs:
             kwargs["A"] = ""
+        # allow for spaces in figure name
+        kwargs["F"] = f'"{kwargs.get("F")}"' if kwargs.get("F") else None
         with Session() as lib:
             lib.call_module("psconvert", build_arg_string(kwargs))
 

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():
@@ -107,6 +108,17 @@ def test_figure_savefig_transparent():
     os.remove(fname)
 
 
+def test_figure_savefig_filename_with_spaces():
+    """
+    Check if savefig (or psconvert) supports filenames with spaces.
+    """
+    fig = Figure()
+    fig.basemap(region=[0, 1, 0, 1], projection="X1c/1c", frame=True)
+    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():
     """
     Check if the arguments being passed to psconvert are correct.