TST: Remove the callback from axes_grid test

greglucas · greglucas · commit e97155b33970 · 2021-05-25T09:20:37.000-06:00
This test puts multiple colorbars into the same cax axes and
some state gets kept due to the callbacks immediately updating
the cax the first time around.
diff --git a/lib/mpl_toolkits/tests/test_axes_grid.py b/lib/mpl_toolkits/tests/test_axes_grid.py
@@ -43,7 +43,10 @@ def test_imagegrid_cbar_mode_edge():
 
         for ax in grid:
             ax.cax.cla()
-            cb = ax.cax.colorbar(ax.images[0])
+            ax.cax.colorbar(ax.images[0])
+            # Remove the callback because we are putting colorbars onto
+            # the same cax
+            ax.images[0].callbacksSM.disconnect(ax.images[0].colorbar_cid)
 
 
 def test_imagegrid():
