diff --git a/test/util/test_cmaps.py b/test/util/test_cmaps.py index b229ecc27..803eb10c0 100644 --- 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)