FIX: Create a new ScalarMappable for axes_grid test

greglucas · greglucas · commit c86c7d10d560 · 2021-05-12T18:20:59.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
@@ -40,15 +40,13 @@ 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()
             # We don't want to share the norms between the colorbar axes
             # so create a deepcopy for each image
-            ax.images[0].norm = copy.deepcopy(ax.images[0].norm)
+            sm = mpl.cm.ScalarMappable(norm=copy.deepcopy(ax.images[0].norm),
+                                       cmap=ax.images[0].get_cmap())
             cb = ax.cax.colorbar(
-                ax.images[0],
+                sm,
                 ticks=mpl.ticker.MaxNLocator(5))  # old default locator.
 
 
