matplotlib
diff --git a/‎lib/matplotlib/figure.py
Copy file name to clipboardExpand all lines: lib/matplotlib/figure.py
+2Lines changed: 2 additions & 0 deletions b/‎lib/matplotlib/figure.py
Copy file name to clipboardExpand all lines: lib/matplotlib/figure.py
+2Lines changed: 2 additions & 0 deletions
diff --git a/‎lib/matplotlib/tests/test_figure.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_figure.py
+11-1Lines changed: 11 additions & 1 deletion b/‎lib/matplotlib/tests/test_figure.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_figure.py
+11-1Lines changed: 11 additions & 1 deletion
@@ -908,6 +908,7 @@ def add_axes(self, *args, **kwargs):
 
         self._axstack.add(key, a)
         self.sca(a)
+        a._remove_method = lambda ax: self.delaxes(ax)
         self.stale = True
         return a
 
@@ -996,6 +997,7 @@ def add_subplot(self, *args, **kwargs):
 
         self._axstack.add(key, a)
         self.sca(a)
+        a._remove_method = lambda ax: self.delaxes(ax)
         self.stale = True
         return a
 
 
@@ -4,7 +4,7 @@
 from matplotlib.externals import six
 from matplotlib.externals.six.moves import xrange
 
-from nose.tools import assert_equal, assert_true, assert_raises
+from nose.tools import assert_equal, assert_true
 from matplotlib.testing.decorators import image_comparison, cleanup
 from matplotlib.axes import Axes
 import matplotlib.pyplot as plt
@@ -181,6 +181,16 @@ def test_set_fig_size():
     assert_equal(fig.get_figheight(), 3)
 
 
+@cleanup
+def test_axes_remove():
+    fig, axes = plt.subplots(2, 2)
+    axes[-1, -1].remove()
+    for ax in axes.ravel()[:-1]:
+        assert ax in fig.axes
+    assert axes[-1, -1] not in fig.axes
+    assert_equal(len(fig.axes), 3)
+
+
 if __name__ == "__main__":
     import nose
     nose.runmodule(argv=['-s', '--with-doctest'], exit=False)