TST: added test of remove functionality

tacaswell · tacaswell · commit 090b6d66b853 · 2015-08-13T22:26:27.000-04:00
diff --git a/lib/matplotlib/tests/test_artist.py b/lib/matplotlib/tests/test_artist.py
@@ -14,6 +14,9 @@
 import matplotlib.collections as mcollections
 from matplotlib.testing.decorators import image_comparison, cleanup
 
+from nose.tools import (assert_true, assert_false, assert_is, assert_in,
+                        assert_not_in)
+
 
 @cleanup
 def test_patch_transform_of_none():
@@ -144,6 +147,30 @@ def test_cull_markers():
     assert len(svg.getvalue()) < 20000
 
 
+@cleanup
+def test_remove():
+    fig, ax = plt.subplots()
+    im = ax.imshow(np.arange(36).reshape(6, 6))
+
+    assert_true(fig.stale)
+    assert_true(ax.stale)
+
+    fig.canvas.draw()
+    assert_false(fig.stale)
+    assert_false(ax.stale)
+
+    assert_in(im, ax.mouseover_set)
+    assert_is(im.axes, ax)
+
+    im.remove()
+
+    assert_is(im.axes, None)
+    assert_is(im.figure, None)
+    assert_not_in(im, ax.mouseover_set)
+    assert_true(fig.stale)
+    assert_true(ax.stale)
+
+
 if __name__ == '__main__':
     import nose
     nose.runmodule(argv=['-s', '--with-doctest'], exit=False)
