Added clim support to tri/tripcolor

andrew-fennell · andrew-fennell · commit 51fff642c250 · 2022-03-30T16:03:18.000-05:00
diff --git a/lib/matplotlib/tests/test_triangulation.py b/lib/matplotlib/tests/test_triangulation.py
@@ -266,6 +266,15 @@ def test_tripcolor_color():
     ax.tripcolor(x, y, facecolors=[1, 2])  # faces
 
 
+def test_tripcolor_clim():
+    (a, b, c) = np.random.rand(10), np.random.rand(10), np.random.rand(10)
+
+    ax = plt.figure().add_subplot()
+    clim = (0.25, 0.75)
+    norm = ax.tripcolor(a, b, c, clim=clim).norm
+    assert((norm.vmin, norm.vmax) == clim)
+
+
 def test_tripcolor_warnings():
     x = [-1, 0, 1, 0]
     y = [0, -1, 0, 1]
diff --git a/lib/matplotlib/tri/tripcolor.py b/lib/matplotlib/tri/tripcolor.py
@@ -120,8 +120,9 @@ def tripcolor(ax, *args, alpha=1.0, norm=None, cmap=None, vmin=None,
             raise ValueError(
                 "shading='gouraud' can only be used when the colors "
                 "are specified at the points, not at the faces.")
-        collection = TriMesh(tri, **kwargs)
-        colors = point_colors
+        _api.check_isinstance((Normalize, None), norm=norm)
+        collection = TriMesh(tri, alpha=alpha, array=point_colors,
+                                cmap=cmap, norm=norm, **kwargs)
     else:
         # Vertices of triangles.
         maskedTris = tri.get_masked_triangles()
@@ -137,13 +138,10 @@ def tripcolor(ax, *args, alpha=1.0, norm=None, cmap=None, vmin=None,
         else:
             colors = facecolors
 
-        collection = PolyCollection(verts, **kwargs)
+        _api.check_isinstance((Normalize, None), norm=norm)
+        collection = PolyCollection(verts, alpha=alpha, array=colors,
+                                    cmap=cmap, norm=norm, **kwargs)
 
-    collection.set_alpha(alpha)
-    collection.set_array(colors)
-    _api.check_isinstance((Normalize, None), norm=norm)
-    collection.set_cmap(cmap)
-    collection.set_norm(norm)
     collection._scale_norm(norm, vmin, vmax)
     ax.grid(False)
 
