test case corrections

prafulgulani · prafulgulani · commit 49a0ab2db7ae · 2025-02-22T15:46:30.000+05:30
diff --git a/lib/matplotlib/tests/test_colors.py b/lib/matplotlib/tests/test_colors.py
@@ -1613,17 +1613,14 @@ def test_set_clim_emits_single_callback():
     callback = unittest.mock.Mock()
     image.norm.callbacks.connect('changed', callback)
 
-    # Initial callback count should be zero
-    assert callback.call_count == 0
+    callback.assert_not_called()
 
     # 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)
