matplotlib
diff --git a/‎lib/matplotlib/tests/baseline_images/test_colorbar/colorbar_norms.png
Copy file name to clipboard
52 KB b/‎lib/matplotlib/tests/baseline_images/test_colorbar/colorbar_norms.png
Copy file name to clipboard
52 KB
diff --git a/‎lib/matplotlib/tests/test_colorbar.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_colorbar.py
+14Lines changed: 14 additions & 0 deletions b/‎lib/matplotlib/tests/test_colorbar.py
Copy file name to clipboardExpand all lines: lib/matplotlib/tests/test_colorbar.py
+14Lines changed: 14 additions & 0 deletions
@@ -570,3 +570,17 @@ def test_colorbar_label():
 
     cbar3 = fig.colorbar(im, orientation='horizontal', label='horizontal cbar')
     assert cbar3.ax.get_xlabel() == 'horizontal cbar'
+
+@image_comparison(
+    baseline_images=['colorbar_norms'], extensions=['png'])
+def test_cbar_norms():
+    # Test colormaps with different norms
+    norms = [mcolors.Normalize(vmin=-1, vmax=2),
+             mcolors.BoundaryNorm(boundaries=[-1, 0, 1], ncolors=4),
+             mcolors.LogNorm(vmin=1, vmax=1e4),
+             mcolors.PowerNorm(gamma=0.2, vmin=0.1, vmax=0.6),
+             mcolors.SymLogNorm(1, vmin=-10, vmax=10),
+             mcolors.TwoSlopeNorm(1, vmin=0, vmax=2)]
+    fig, axs = plt.subplots(ncols=len(norms), constrained_layout=True)
+    for ax, norm in zip(axs, norms):
+        fig.colorbar(cm.ScalarMappable(norm=norm, cmap='viridis'), cax=ax, extend='both')