TST: tweak tests so they run

tacaswell · tacaswell · commit b37f14214eda · 2017-07-08T22:22:50.000-04:00
diff --git a/lib/matplotlib/tests/test_backend_pdf.py b/lib/matplotlib/tests/test_backend_pdf.py
@@ -195,15 +195,15 @@ def psfont(*args, **kwargs):
         fig.savefig(tmpfile, format='pdf')
 
 
-def test_pdf_savefig_when_color_is_none(tempdir_factory):
-    rcParams['_internal.classic_mode'] = False
-    plt.subplot()
+@pytest.mark.style('default')
+def test_pdf_savefig_when_color_is_none(tmpdir_factory):
+    fig, ax = plt.subplots()
     plt.axis('off')
-    plt.plot(np.sin(np.linspace(-5, 5, 100)), 'v', c='none')
+    ax.plot(np.sin(np.linspace(-5, 5, 100)), 'v', c='none')
     tmpdir_name = str(np.random.randint(100000, 10000000))
-    actual_image = tempdir_factory.mktemp(tmpdir_name).join('figure.pdf')
-    expected_image = tempdir_factory.mktemp(tmpdir_name).join('figure.eps')
-    plt.savefig(str(actual_image), format='pdf')
-    plt.savefig(str(expected_image), format='eps')
+    actual_image = tmpdir_factory.mktemp(tmpdir_name).join('figure.pdf')
+    expected_image = tmpdir_factory.mktemp(tmpdir_name).join('figure.eps')
+    fig.savefig(str(actual_image), format='pdf')
+    fig.savefig(str(expected_image), format='eps')
     result = compare_images(str(actual_image), str(expected_image), 0)
     assert result is None
