Merge pull request matplotlib#6754 from Kojoley/fix-error-handling-in…

…-imagecomparisontest Fixed error handling in `ImageComparisonTest.setup_class`
folk85 · Jul 16, 2016 · 416c2a2 · 416c2a2
1 parent d7af968
commit 416c2a2
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/lib/matplotlib/testing/decorators.py b/lib/matplotlib/testing/decorators.py
@@ -181,17 +181,14 @@ class ImageComparisonTest(CleanupTest):
     @classmethod
     def setup_class(cls):
         CleanupTest.setup_class()
-        cls._initial_settings = mpl.rcParams.copy()
         try:
             matplotlib.style.use(cls._style)
+            matplotlib.testing.set_font_settings_for_testing()
+            cls._func()
         except:
             # Restore original settings before raising errors during the update.
-            mpl.rcParams.clear()
-            mpl.rcParams.update(cls._initial_settings)
+            CleanupTest.teardown_class()
             raise
-        cls.original_settings = cls._initial_settings
-        matplotlib.testing.set_font_settings_for_testing()
-        cls._func()
 
     @classmethod
     def teardown_class(cls):