use figure_context for tests (#37)

Author: mathause

mplotutils/tests/__init__.py

@@ -0,0 +1,23 @@
+import contextlib
+
+import matplotlib.pyplot as plt
+
+
+@contextlib.contextmanager
+def figure_context(*args, **kwargs):
+    fig = plt.figure(*args, **kwargs)
+
+    try:
+        yield fig
+    finally:
+        plt.close(fig)
+
+
+@contextlib.contextmanager
+def subplots_context(*args, **kwargs):
+    fig, axs = plt.subplots(*args, **kwargs)
+
+    try:
+        yield fig, axs
+    finally:
+        plt.close(fig)