test for issue matplotlib#1683

tacaswell · tacaswell · commit 75a92df91a05 · 2013-01-25T22:09:10.000-06:00
diff --git a/lib/matplotlib/tests/test_backend_qt4.py b/lib/matplotlib/tests/test_backend_qt4.py
@@ -1,10 +1,25 @@
 from matplotlib import rcParams
 from matplotlib import pyplot as plt
 from matplotlib.testing.decorators import cleanup
+from matplotlib._pylab_helpers import Gcf
+import copy
+
 
 @cleanup
 def test_fig_close():
-    rcParams['backend'] = 'qt4agg'
+    # force switch to the Qt4 backend
+    plt.switch_backend('Qt4Agg')
+
+    #save the state of Gcf.figs
+    init_figs = copy.copy(Gcf.figs)
 
+    # make a figure using pyplot interface
     fig = plt.figure()
-    fig.close()
+
+    # simulate user clicking the close button by reaching in
+    # and calling close on the underlying Qt object
+    fig.canvas.manager.window.close()
+
+    # assert that we have removed the reference to the FigureManager
+    # that got added by plt.figure()
+    assert(init_figs == Gcf.figs)
