TST: Update unittests

openradar · Nov 6, 2023 · 4b21cba · 4b21cba
1 parent 9faadd3
commit 4b21cba
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/test_cm_colorblind.py b/tests/test_cm_colorblind.py
@@ -19,6 +19,8 @@ def test_colormaps_exist():
     assert isinstance(cm_colorblind.CM_depol_r, matplotlib.colors.Colormap)
     assert isinstance(cm_colorblind.CM_rhohv, matplotlib.colors.Colormap)
     assert isinstance(cm_colorblind.CM_rhohv_r, matplotlib.colors.Colormap)
+    assert isinstance(cm_colorblind.plasmidis, matplotlib.colors.Colormap)
+    assert isinstance(cm_colorblind.plasmidis_r, matplotlib.colors.Colormap)
 
 
 def test_colormaps_registered():
@@ -57,3 +59,9 @@ def test_colormaps_registered():
 
     cmap = matplotlib.colormaps.get_cmap('CM_rhohv_r')
     assert isinstance(cmap, matplotlib.colors.Colormap)
+
+    cmap = matplotlib.colormaps.get_cmap('plasmidis')
+    assert isinstance(cmap, matplotlib.colors.Colormap)
+
+    cmap = matplotlib.colormaps.get_cmap('plasmidis_r')
+    assert isinstance(cmap, matplotlib.colors.Colormap)