Merge pull request #6022 from anntzer/cleanup-generative-tests-again

tacaswell · tacaswell · commit 80be85391bbe · 2016-03-26T18:34:12.000-04:00
Make @cleanup *really* support generative tests.
diff --git a/lib/matplotlib/testing/decorators.py b/lib/matplotlib/testing/decorators.py
@@ -130,7 +130,7 @@ def cleanup(style=None):
     # writing a decorator with optional arguments.
 
     def make_cleanup(func):
-        if inspect.isgenerator(func):
+        if inspect.isgeneratorfunction(func):
             @functools.wraps(func)
             def wrapped_callable(*args, **kwargs):
                 original_units_registry = matplotlib.units.registry.copy()
