Update test_plot.py

added testcase for pydata#7014
veenstrajelmer · Feb 23, 2023 · 3d84b2c · 3d84b2c
1 parent 1de881e
commit 3d84b2c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/xarray/tests/test_plot.py b/xarray/tests/test_plot.py
@@ -1193,6 +1193,11 @@ def test_discrete_colormap_provided_boundary_norm(self) -> None:
         norm = mpl.colors.BoundaryNorm([0, 5, 10, 15], 4)
         primitive = self.darray.plot.contourf(norm=norm)
         np.testing.assert_allclose(primitive.levels, norm.boundaries)
+
+    def test_discrete_colormap_provided_boundary_norm_matching_cmap_levels(self):
+        norm = mpl.colors.BoundaryNorm([0, 5, 10, 15], 4)
+        primitive = self.darray.plot.contourf(norm=norm)
+        assert primitive.colorbar.norm.Ncmap+1 == primitive.colorbar.norm.N
 
 
 class Common2dMixin: