TST: use pytest tmpdir fixture instead of tempdir + shutil

tacaswell · tacaswell · commit b9d944b376fd · 2017-10-23T15:23:06.000-04:00
diff --git a/lib/matplotlib/tests/test_png.py b/lib/matplotlib/tests/test_png.py
@@ -5,14 +5,12 @@
 from six import BytesIO
 import glob
 import os
-import shutil
 import numpy as np
 import pytest
 
 from matplotlib.testing.decorators import image_comparison
 from matplotlib import pyplot as plt
 import matplotlib.cm as cm
-import tempfile
 import sys
 on_win = (sys.platform == 'win32')
 
@@ -49,8 +47,8 @@ def test_imread_png_uint16():
     assert np.sum(img.flatten()) == 134184960
 
 
-def test_truncated_file():
-    d = tempfile.mkdtemp()
+def test_truncated_file(tmpdir):
+    d = tmpdir.mkdir('test')
     fname = os.path.join(d, 'test.png')
     fname_t = os.path.join(d, 'test_truncated.png')
     plt.savefig(fname)
@@ -62,8 +60,6 @@ def test_truncated_file():
     with pytest.raises(Exception):
         plt.imread(fname_t)
 
-    shutil.rmtree(d)
-
 
 def test_truncated_buffer():
     b = BytesIO()
