In unit/memleak, write to in-memory buffer instead of file.

anntzer · anntzer · commit 4886ea9150c5 · 2017-12-03T12:51:35.000-08:00
(Avoids cluttering the source directory.)
diff --git a/unit/memleak.py b/unit/memleak.py
@@ -3,6 +3,7 @@
 from __future__ import print_function
 
 import gc
+from io import BytesIO
 
 try:
     import tracemalloc
@@ -110,7 +111,7 @@ def __call__(self):
             ax = fig.add_subplot(224)
             ax.pcolor(10 * np.random.rand(50, 50))
 
-        fig.savefig('tmp', dpi=75)
+        fig.savefig(BytesIO(), dpi=75)
         plt.close(1)
 
 
