Ensure pyplot wrapper-funcs return result

Author: lukelbd

proplot/ui.py

@@ -20,12 +20,12 @@
 __all__ = [
     'subplots',
     'SubplotsContainer',
+    'show',
     'close',
+    'switch_backend',
     'ion',
     'ioff',
     'isinteractive',
-    'show',
-    'switch_backend',
 ]
 
 docstring.snippets['pyplot.statement'] = """
@@ -54,72 +54,72 @@
 
 
 @docstring.add_snippets
-def close(*args, **kwargs):
+def show(*args, **kwargs):
     """
-    Call `matplotlib.pyplot.close`.
+    Call `matplotlib.pyplot.show`.
     %(pyplot.statement)s
 
     Parameters
     ----------
     *args, **kwargs
-        Passed to `matplotlib.pyplot.close`.
+        Passed to `matplotlib.pyplot.show`.
     """
-    plt.close(*args, **kwargs)
+    return plt.show(*args, **kwargs)
 
 
 @docstring.add_snippets
-def ion():
+def close(*args, **kwargs):
     """
-    Call `matplotlib.pyplot.ion`.
+    Call `matplotlib.pyplot.close`.
     %(pyplot.statement)s
+
+    Parameters
+    ----------
+    *args, **kwargs
+        Passed to `matplotlib.pyplot.close`.
     """
-    plt.ion()
+    return plt.close(*args, **kwargs)
 
 
 @docstring.add_snippets
-def ioff():
+def switch_backend(*args, **kwargs):
     """
-    Call `matplotlib.pyplot.ioff`.
+    Call `matplotlib.pyplot.switch_backend`.
     %(pyplot.statement)s
+
+    Parameters
+    ----------
+    *args, **kwargs
+        Passed to `matplotlib.pyplot.switch_backend`.
     """
-    plt.ioff()
+    return plt.switch_backend(*args, **kwargs)
 
 
 @docstring.add_snippets
-def isinteractive():
+def ion():
     """
-    Call `matplotlib.pyplot.isinteractive`.
+    Call `matplotlib.pyplot.ion`.
     %(pyplot.statement)s
     """
-    plt.isinteractive()
+    return plt.ion()
 
 
 @docstring.add_snippets
-def show(*args, **kwargs):
+def ioff():
     """
-    Call `matplotlib.pyplot.show`.
+    Call `matplotlib.pyplot.ioff`.
     %(pyplot.statement)s
-
-    Parameters
-    ----------
-    *args, **kwargs
-        Passed to `matplotlib.pyplot.show`.
     """
-    plt.show(*args, **kwargs)
+    return plt.ioff()
 
 
 @docstring.add_snippets
-def switch_backend(*args, **kwargs):
+def isinteractive():
     """
-    Call `matplotlib.pyplot.switch_backend`.
+    Call `matplotlib.pyplot.isinteractive`.
     %(pyplot.statement)s
-
-    Parameters
-    ----------
-    *args, **kwargs
-        Passed to `matplotlib.pyplot.switch_backend`.
     """
-    plt.switch_backend(*args, **kwargs)
+    return plt.isinteractive()
 
 
 def _journals(journal):