Merge pull request #17617 from anntzer/figeq

timhoffm · web-flow · commit 45d38777ed1d · 2020-06-17T11:25:24.000+02:00
Rewrite pdf test to use check_figures_equal.
diff --git a/lib/matplotlib/tests/test_backend_pdf.py b/lib/matplotlib/tests/test_backend_pdf.py
@@ -11,8 +11,7 @@
 import matplotlib as mpl
 from matplotlib import dviread, pyplot as plt, checkdep_usetex, rcParams
 from matplotlib.backends.backend_pdf import PdfPages
-from matplotlib.testing.compare import compare_images
-from matplotlib.testing.decorators import image_comparison
+from matplotlib.testing.decorators import check_figures_equal, image_comparison
 
 
 needs_usetex = pytest.mark.skipif(
@@ -245,16 +244,13 @@ def psfont(*args, **kwargs):
 
 
 @pytest.mark.style('default')
-def test_pdf_savefig_when_color_is_none(tmpdir):
-    fig, ax = plt.subplots()
-    plt.axis('off')
-    ax.plot(np.sin(np.linspace(-5, 5, 100)), 'v', c='none')
-    actual_image = tmpdir.join('figure.pdf')
-    expected_image = tmpdir.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
+@check_figures_equal(extensions=["pdf", "eps"])
+def test_pdf_eps_savefig_when_color_is_none(fig_test, fig_ref):
+    ax_test = fig_test.add_subplot()
+    ax_test.set_axis_off()
+    ax_test.plot(np.sin(np.linspace(-5, 5, 100)), "v", c="none")
+    ax_ref = fig_ref.add_subplot()
+    ax_ref.set_axis_off()
 
 
 @needs_usetex
