Move cleanup to the individual checks

blink1073 · blink1073 · commit d5b2d2f2d372 · 2014-11-09T18:52:16.000-06:00
diff --git a/lib/matplotlib/tests/test_widgets.py b/lib/matplotlib/tests/test_widgets.py
@@ -63,6 +63,7 @@ def get_event(ax, button=1, xdata=0, ydata=0, key=None, step=1):
     return event
 
 
+@cleanup
 def check_rectangle(**kwargs):
     ax = plt.gca()
     ax.plot([0, 200], [0, 200])
@@ -87,7 +88,6 @@ def onselect(epress, erelease):
     assert ax._got_onselect
 
 
-@cleanup
 def test_rectangle_selector():
     check_rectangle()
     check_rectangle(drawtype='line', useblit=False)
@@ -97,6 +97,7 @@ def test_rectangle_selector():
     check_rectangle(rectprops=dict(fill=True))
 
 
+@cleanup
 def check_span(*args, **kwargs):
     ax = plt.gca()
     ax.plot([0, 200], [0, 200])
@@ -130,13 +131,13 @@ def onmove(vmin, vmax):
         assert ax._got_on_move
 
 
-@cleanup
 def test_span_selector():
     check_span('horizontal', minspan=10, useblit=True)
     check_span('vertical', onmove_callback=True, button=1)
     check_span('horizontal', rectprops=dict(fill=True))
 
 
+@cleanup
 def check_lasso_selector(**kwargs):
     ax = plt.gca()
     ax.plot([0, 200], [0, 200])
@@ -159,7 +160,6 @@ def onselect(verts):
     assert ax._got_onselect
 
 
-@cleanup
 def test_lasso_selector():
     check_lasso_selector()
     check_lasso_selector(useblit=False, lineprops=dict(color='red'))
