Restore ImageComparisonTest from ImageComparisonDecorator.

QuLogic · QuLogic · commit 0ab7f4b85212 · 2017-03-29T21:07:24.000-04:00
diff --git a/lib/matplotlib/testing/decorators.py b/lib/matplotlib/testing/decorators.py
@@ -281,7 +281,7 @@ def compare(self, idx, baseline, extension):
         _raise_on_image_difference(expected_fname, actual_fname, self.tol)
 
 
-class ImageComparisonDecorator(CleanupTest, _ImageComparisonBase):
+class ImageComparisonTest(CleanupTest, _ImageComparisonBase):
     def __init__(self, baseline_images, extensions, tol,
                  freetype_version, remove_text, savefig_kwargs, style):
         _ImageComparisonBase.__init__(self, tol, remove_text, savefig_kwargs)
@@ -436,7 +436,7 @@ def image_comparison(baseline_images, extensions=None, tol=0,
         if baseline_images is None:
             raise ValueError('baseline_images must be specified')
 
-        return ImageComparisonDecorator(
+        return ImageComparisonTest(
             baseline_images=baseline_images, extensions=extensions, tol=tol,
             freetype_version=freetype_version, remove_text=remove_text,
             savefig_kwargs=savefig_kwarg, style=style)
