FIX: check for string color in ContourSet

rcomer · rcomer · commit 2b2f19575aec · 2024-09-07T16:04:17.000+01:00
diff --git a/lib/matplotlib/contour.py b/lib/matplotlib/contour.py
@@ -702,6 +702,11 @@ def __init__(self, ax, *args,
         self._extend_min = self.extend in ['min', 'both']
         self._extend_max = self.extend in ['max', 'both']
         if self.colors is not None:
+            if isinstance(self.colors, str):
+                colors = [colors]
+            else:
+                colors = colors
+
             ncolors = len(self.levels)
             if self.filled:
                 ncolors -= 1
@@ -718,19 +723,19 @@ def __init__(self, ax, *args,
             total_levels = (ncolors +
                             int(self._extend_min) +
                             int(self._extend_max))
-            if (len(self.colors) == total_levels and
+            if (len(colors) == total_levels and
                     (self._extend_min or self._extend_max)):
                 use_set_under_over = True
                 if self._extend_min:
                     i0 = 1
 
-            cmap = mcolors.ListedColormap(self.colors[i0:None], N=ncolors)
+            cmap = mcolors.ListedColormap(colors[i0:None], N=ncolors)
 
             if use_set_under_over:
                 if self._extend_min:
-                    cmap.set_under(self.colors[0])
+                    cmap.set_under(colors[0])
                 if self._extend_max:
-                    cmap.set_over(self.colors[-1])
+                    cmap.set_over(colors[-1])
 
         # label lists must be initialized here
         self.labelTexts = []
diff --git a/lib/matplotlib/tests/test_contour.py b/lib/matplotlib/tests/test_contour.py
@@ -171,6 +171,17 @@ def test_given_colors_levels_and_extends():
         plt.colorbar(c, ax=ax)
 
 
+def test_single_color_and_extend():
+    z = [[0, 1], [1, 2]]
+    color = 'green'
+    levels = [0.5, 1, 1.5]
+
+    _, ax = plt.subplots()
+    cs = ax.contour(z, levels=levels, colors=color, extend='both')
+    for c in cs.get_edgecolors():
+        assert same_color(c, color)
+
+
 @image_comparison(['contour_log_locator.svg'], style='mpl20', remove_text=False)
 def test_log_locator_levels():
 
