matplotlib
diff --git a/‎lib/matplotlib/contour.py
Copy file name to clipboardExpand all lines: lib/matplotlib/contour.py
+9-4Lines changed: 9 additions & 4 deletions b/‎lib/matplotlib/contour.py
Copy file name to clipboardExpand all lines: lib/matplotlib/contour.py
+9-4Lines changed: 9 additions & 4 deletions
diff --git a/‎lib/matplotlib/tests/test_contour.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_contour.py
+11Lines changed: 11 additions & 0 deletions b/‎lib/matplotlib/tests/test_contour.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_contour.py
+11Lines changed: 11 additions & 0 deletions
@@ -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 = []
 
@@ -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():