Relocated test_set_clim_emits_single_callback to test_colors.py

prafulgulani · prafulgulani · commit bf5370d8d97f · 2025-02-12T21:51:45.000+05:30
diff --git a/lib/matplotlib/colorizer.py b/lib/matplotlib/colorizer.py
@@ -279,9 +279,9 @@ def set_clim(self, vmin=None, vmax=None):
             if vmax is not None:
                 self.norm.vmax = colors._sanitize_extrema(vmax)
 
-        # self.changed() will now emit a update signal if the limits are changed
+        # emit a update signal if the limits are changed
         if orig_vmin_vmax != (self.norm.vmin, self.norm.vmax):
-            self.changed()
+            self.norm.callbacks.process('changed')
 
     def get_clim(self):
         """
diff --git a/lib/matplotlib/tests/test_colorbar.py b/lib/matplotlib/tests/test_colorbar.py
@@ -18,8 +18,6 @@
 from matplotlib.ticker import FixedLocator, LogFormatter, StrMethodFormatter
 from matplotlib.testing.decorators import check_figures_equal
 
-import unittest.mock
-
 
 def _get_cmap_norms():
     """
@@ -662,26 +660,6 @@ def test_colorbar_scale_reset():
     assert pcm.norm.vmax == z.max()
 
 
-def test_set_clim_emits_single_callback():
-    data = np.array([[1, 2], [3, 4]])
-    fig, ax = plt.subplots()
-    image = ax.imshow(data, cmap='viridis')
-
-    callback = unittest.mock.Mock()
-    image.norm.callbacks.connect('changed', callback)
-
-    # Initial callback count should be zero
-    assert callback.call_count == 0
-
-    # Call set_clim() to update the limits
-    image.set_clim(1, 5)
-
-    # Assert that only one "changed" callback is sent after calling set_clim()
-    callback.assert_called_once()
-
-    plt.close(fig)
-
-
 def test_colorbar_get_ticks_2():
     plt.rcParams['_internal.classic_mode'] = False
     fig, ax = plt.subplots()
diff --git a/lib/matplotlib/tests/test_colors.py b/lib/matplotlib/tests/test_colors.py
@@ -1605,6 +1605,26 @@ def test_norm_deepcopy():
     assert norm2.vmin == norm.vmin
 
 
+def test_set_clim_emits_single_callback():
+    data = np.array([[1, 2], [3, 4]])
+    fig, ax = plt.subplots()
+    image = ax.imshow(data, cmap='viridis')
+
+    callback = unittest.mock.Mock()
+    image.norm.callbacks.connect('changed', callback)
+
+    # Initial callback count should be zero
+    assert callback.call_count == 0
+
+    # Call set_clim() to update the limits
+    image.set_clim(1, 5)
+
+    # Assert that only one "changed" callback is sent after calling set_clim()
+    callback.assert_called_once()
+
+    plt.close(fig)
+
+
 def test_norm_callback():
     increment = unittest.mock.Mock(return_value=None)
 
