Add test cleanup decorator

blink1073 · blink1073 · commit cf668122e02b · 2014-11-12T05:43:22.000-06:00
diff --git a/lib/matplotlib/tests/test_widgets.py b/lib/matplotlib/tests/test_widgets.py
@@ -9,6 +9,7 @@
 
 import matplotlib.widgets as widgets
 import matplotlib.pyplot as plt
+from matplotlib.testing.decorators import cleanup
 
 
 def get_event(ax, button=1, xdata=0, ydata=0, key=None, step=1):
@@ -86,6 +87,7 @@ def onselect(epress, erelease):
     assert ax._got_onselect
 
 
+@cleanup
 def test_rectangle_selector():
     check_rectangle()
     check_rectangle(drawtype='line', useblit=False)
@@ -128,6 +130,7 @@ 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)
@@ -156,6 +159,7 @@ def onselect(verts):
     assert ax._got_onselect
 
 
+@cleanup
 def test_lasso_selector():
     check_lasso_selector()
     check_lasso_selector(useblit=False, lineprops=dict(color='red'))
