Fix bug that prevented boolean to -A from working

sixy6e · Feb 12, 2021 · eadb847 · eadb847
1 parent d3c0a50
commit eadb847
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/pygmt/src/subplot.py b/pygmt/src/subplot.py
@@ -142,7 +142,8 @@ def subplot(self, nrows=1, ncols=1, **kwargs):
     {XY}
     """
     kwargs = self._preprocess(**kwargs)  # pylint: disable=protected-access
-    kwargs["A"] = f'"{kwargs.get("A")}"' if kwargs.get("A") else None
+    # allow for spaces in string with needing double quotes
+    kwargs["A"] = f'"{kwargs.get("A")}"' if kwargs.get("A") is not None else None
     kwargs["T"] = f'"{kwargs.get("T")}"' if kwargs.get("T") else None
 
     with Session() as lib:
@@ -205,7 +206,8 @@ def set_panel(self, panel=None, **kwargs):
     {V}
     """
     kwargs = self._preprocess(**kwargs)  # pylint: disable=protected-access
-    kwargs["A"] = f'"{kwargs.get("A")}"' if kwargs.get("A") else None
+    # allow for spaces in string with needing double quotes
+    kwargs["A"] = f'"{kwargs.get("A")}"' if kwargs.get("A") is not None else None
     # convert tuple or list to comma-separated str
     panel = ",".join(map(str, panel)) if is_nonstr_iter(panel) else panel
 

diff --git a/pygmt/tests/test_subplot.py b/pygmt/tests/test_subplot.py
@@ -48,12 +48,12 @@ def test_subplot_autolabel_margins_title():
     fig_ref, fig_test = Figure(), Figure()
     kwargs = dict(nrows=2, ncols=1, figsize=("15c", "6c"))
 
-    with fig_ref.subplot(A="(1)", M="0.3c/0.1c", T="Subplot Title", **kwargs):
+    with fig_ref.subplot(A="a)", M="0.3c/0.1c", T="Subplot Title", **kwargs):
         fig_ref.basemap(region=[0, 1, 2, 3], frame="WSne", c="0,0")
         fig_ref.basemap(region=[4, 5, 6, 7], frame="WSne", c="1,0")
 
     with fig_test.subplot(
-        autolabel="(1)", margins=["0.3c", "0.1c"], title="Subplot Title", **kwargs
+        autolabel=True, margins=["0.3c", "0.1c"], title="Subplot Title", **kwargs
     ):
         fig_test.basemap(region=[0, 1, 2, 3], frame="WSne", panel=[0, 0])
         fig_test.basemap(region=[4, 5, 6, 7], frame="WSne", panel=[1, 0])