add a quick test

NCAR · Apr 19, 2024 · 02e09e2 · 02e09e2
1 parent f75859d
commit 02e09e2
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_util.py b/tests/test_util.py
@@ -1 +1,13 @@
 import pytest
+import matplotlib as mpl
+
+from geocat.viz.util import truncate_colormap
+
+
+def test_truncate_colormap():
+    cmap = mpl.colormaps['terrain']
+    truncated_cmap = truncate_colormap(cmap, 0.1, 0.9)
+
+    assert isinstance(truncated_cmap, mpl.colors.LinearSegmentedColormap)
+    assert truncated_cmap(0.0) == cmap(0.1)
+    assert truncated_cmap(1.0) == cmap(0.9)