FIX: Update AxesGrid colorbar test

greglucas · greglucas · commit 29ae631cd2b4 · 2021-03-03T21:17:41.000-07:00
This test clears out a previous colorbar axis and the
norms were getting shared and updated between these images.
To fix this, we can deepcopy the norm before creating the colorbar.
diff --git a/lib/mpl_toolkits/tests/test_axes_grid.py b/lib/mpl_toolkits/tests/test_axes_grid.py
@@ -1,3 +1,4 @@
+import copy
 import numpy as np
 
 import matplotlib as mpl
@@ -43,6 +44,9 @@ def test_imagegrid_cbar_mode_edge():
         # "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)
             cb = ax.cax.colorbar(
                 ax.images[0],
                 ticks=mpl.ticker.MaxNLocator(5))  # old default locator.
