Use image_comparison decorator in pgf tests.

anntzer · anntzer · commit bbbfef680092 · 2018-01-28T14:33:50.000-08:00
diff --git a/lib/matplotlib/tests/test_backend_pgf.py b/lib/matplotlib/tests/test_backend_pgf.py
@@ -11,7 +11,7 @@
 import matplotlib.pyplot as plt
 from matplotlib.compat import subprocess
 from matplotlib.testing.compare import compare_images, ImageComparisonFailure
-from matplotlib.testing.decorators import _image_directories
+from matplotlib.testing.decorators import image_comparison, _image_directories
 
 baseline_dir, result_dir = _image_directories(lambda: 'dummy func')
 
@@ -81,20 +81,21 @@ def create_figure():
 
 # test compiling a figure to pdf with xelatex
 @needs_xelatex
-@pytest.mark.style('default')
 @pytest.mark.backend('pgf')
+@image_comparison(baseline_images=['pgf_xelatex'], extensions=['pdf'],
+                  style='default')
 def test_xelatex():
     rc_xelatex = {'font.family': 'serif',
                   'pgf.rcfonts': False}
     mpl.rcParams.update(rc_xelatex)
     create_figure()
-    compare_figure('pgf_xelatex.pdf', tol=0)
 
 
 # test compiling a figure to pdf with pdflatex
 @needs_pdflatex
-@pytest.mark.style('default')
 @pytest.mark.backend('pgf')
+@image_comparison(baseline_images=['pgf_pdflatex'], extensions=['pdf'],
+                  style='default')
 def test_pdflatex():
     import os
     if os.environ.get('APPVEYOR', False):
@@ -108,7 +109,6 @@ def test_pdflatex():
                                     '\\usepackage[T1]{fontenc}']}
     mpl.rcParams.update(rc_pdflatex)
     create_figure()
-    compare_figure('pgf_pdflatex.pdf', tol=0)
 
 
 # test updating the rc parameters for each figure
@@ -162,8 +162,9 @@ def test_pathclip():
 
 # test mixed mode rendering
 @needs_xelatex
-@pytest.mark.style('default')
 @pytest.mark.backend('pgf')
+@image_comparison(baseline_images=['pgf_mixedmode'], extensions=['pdf'],
+                  style='default')
 def test_mixedmode():
     rc_xelatex = {'font.family': 'serif',
                   'pgf.rcfonts': False}
@@ -172,7 +173,6 @@ def test_mixedmode():
     Y, X = np.ogrid[-1:1:40j, -1:1:40j]
     plt.figure()
     plt.pcolor(X**2 + Y**2).set_rasterized(True)
-    compare_figure('pgf_mixedmode.pdf', tol=0)
 
 
 # test bbox_inches clipping
