Fixed some issues in color mapping introduced after code refactoring

xcube-dev · Jul 19, 2024 · d4d7152 · d4d7152
1 parent 2aa8eb1
commit d4d7152
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/util/test_cmaps.py b/test/util/test_cmaps.py
@@ -112,6 +112,7 @@ def test_get_cmap_from_code_type_key(self):
         self.assertIsInstance(cmap(np.linspace(0, 1, 10)), np.ndarray)
         self.assertIsInstance(colormap, Colormap)
         self.assertEqual("ucb783473", colormap.cm_name)
+        self.assertEqual("key", colormap.cm_type)
         self.assertEqual(CUSTOM_CATEGORY.name, colormap.cat_name)
         self.assertEqual([1, 2, 3, 5, 6], colormap.values)
 
@@ -129,6 +130,7 @@ def test_get_cmap_from_code_type_bound(self):
         self.assertIsInstance(cmap(np.linspace(0, 1, 10)), np.ndarray)
         self.assertIsInstance(colormap, Colormap)
         self.assertEqual("ucb783474", colormap.cm_name)
+        self.assertEqual("bound", colormap.cm_type)
         self.assertEqual(CUSTOM_CATEGORY.name, colormap.cat_name)
         self.assertEqual((0.0, 0.6, 1.0), colormap.values)
 
@@ -146,6 +148,7 @@ def test_get_cmap_from_code_reversed_alpha(self):
         self.assertIsInstance(cmap(np.linspace(0, 1, 10)), np.ndarray)
         self.assertIsInstance(colormap, Colormap)
         self.assertEqual("ucb783475", colormap.cm_name)
+        self.assertEqual("node", colormap.cm_type)
         self.assertEqual(CUSTOM_CATEGORY.name, colormap.cat_name)
         self.assertEqual((0.0, 0.5, 1.0), colormap.values)
 
@@ -157,6 +160,7 @@ def test_get_cmap_from_code_invalid(self):
         self.assertIsInstance(cmap, matplotlib.colors.LinearSegmentedColormap)
         self.assertIsInstance(colormap, Colormap)
         self.assertEqual("Reds", colormap.cm_name)
+        self.assertEqual("node", colormap.cm_type)
         self.assertEqual("Sequential", colormap.cat_name)
         self.assertIsNone(colormap.values)