TST: add test for min/max extents

jklymak · jklymak · commit 537946e4382c · 2021-05-19T10:03:54.000-07:00
diff --git a/lib/matplotlib/tests/test_colorbar.py b/lib/matplotlib/tests/test_colorbar.py
@@ -457,11 +457,12 @@ def test_colorbar_get_ticks():
     assert len(cbar.get_ticks(minor=True)) == 0
 
 
-def test_colorbar_lognorm_extension():
+@pytest.mark.parametrize("extend", ['both', 'min', 'max'])
+def test_colorbar_lognorm_extension(extend):
     # Test that colorbar with lognorm is extended correctly
     f, ax = plt.subplots()
     cb = ColorbarBase(ax, norm=LogNorm(vmin=0.1, vmax=1000.0),
-                      orientation='vertical', extend='both')
+                      orientation='vertical', extend=extend)
     assert cb._values[0] >= 0.0
 
 
