From 02e09e2daaf98afadc9c27b4e1e31ec94c195e6e Mon Sep 17 00:00:00 2001
From: Katelyn FitzGerald <7872563+kafitzgerald@users.noreply.github.com>
Date: Fri, 19 Apr 2024 11:45:05 -0600
Subject: [PATCH] add a quick test

---
 tests/test_util.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/tests/test_util.py b/tests/test_util.py
index 5871ed8..8084bec 100644
--- 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)