FIX: Create a new ScalarMappable for axes_grid test

greglucas · greglucas · commit 17b9000ba01c · 2021-05-24T20:15:28.000-06:00
There is some shared state when adding colorbars to the axes_grid
test here due to competing calls with colorbar.update_normal. To
avoid that, we create a new ScalarMappable instance that creates
a new norm that doesn't share state.
diff --git a/lib/mpl_toolkits/tests/test_axes_grid.py b/lib/mpl_toolkits/tests/test_axes_grid.py
@@ -41,8 +41,6 @@ def test_imagegrid_cbar_mode_edge():
         ax3.imshow(np.abs(arr), cmap='jet')
         ax4.imshow(np.arctan2(arr.imag, arr.real), cmap='hsv')
 
-        # In each row/column, the "first" colorbars must be overwritten by the
-        # "second" ones.  To achieve this, clear out the axes first.
         for ax in grid:
             ax.cax.cla()
             cb = ax.cax.colorbar(ax.images[0])
