TST: Adding tests for clip and vcenter to norm updater

greglucas · greglucas · commit b26748d22d3c · 2021-03-22T19:56:02.000-06:00
diff --git a/lib/matplotlib/colors.py b/lib/matplotlib/colors.py
@@ -1439,7 +1439,9 @@ def vcenter(self):
 
     @vcenter.setter
     def vcenter(self, vcenter):
-        self._vcenter = vcenter
+        if vcenter != self._vcenter:
+            self._vcenter = vcenter
+            self._changed()
         if self.vmax is not None:
             # recompute halfrange assuming vmin and vmax represent
             # min and max of data
diff --git a/lib/matplotlib/tests/test_colors.py b/lib/matplotlib/tests/test_colors.py
@@ -1417,3 +1417,17 @@ def test_scalarmappable_norm_update():
     sm.stale = False
     norm.vmax = 5
     assert sm.stale
+    sm.stale = False
+    norm.clip = True
+    assert sm.stale
+    # change to the CenteredNorm and TwoSlopeNorm to test those
+    norm = mcolors.CenteredNorm()
+    sm = matplotlib.cm.ScalarMappable(norm=norm, cmap='plasma')
+    sm.stale = False
+    norm.vcenter = 1
+    assert sm.stale
+    norm = mcolors.TwoSlopeNorm(vcenter=0, vmin=-1, vmax=1)
+    sm = matplotlib.cm.ScalarMappable(norm=norm, cmap='plasma')
+    sm.stale = False
+    norm.vcenter = 1
+    assert sm.stale
