play with listed thresh

Author: cvanelteren

ultraplot/tests/test_colors.py

@@ -2,7 +2,7 @@
 import pytest
 import numpy as np
 import matplotlib.colors as mcolors
-
+import ultraplot as uplt
 from ultraplot import colors as pcolors
 from ultraplot import config
 
@@ -22,24 +22,31 @@ def setup_teardown():
 
 def test_lazy_loading_builtin():
     """
-    Test that built-in colormaps are lazy-loaded.
+    Test that built-in colormaps are lazy-loaded and translated correctly.
     """
-    pcolors._cmap_database = pcolors._init_cmap_database()  # ensure pristine state
-    # Before access, it should be a matplotlib colormap
-    cmap_raw = pcolors._cmap_database._cmaps["viridis"]
-    assert not isinstance(
-        cmap_raw, (pcolors.ContinuousColormap, pcolors.DiscreteColormap)
-    )
-
-    # After access, it should be an ultraplot colormap
-    cmap_get = pcolors._cmap_database.get_cmap("viridis")
-    print(type(cmap_get))  # is continuous
-    assert isinstance(cmap_get, pcolors.ContinuousColormap)
-
-    # The internal representation should also be updated
-    cmap_raw_after = pcolors._cmap_database._cmaps["viridis"]
-    print(type(cmap_raw_after))  # is continuous
-    assert isinstance(cmap_raw_after, pcolors.DiscreteColormap)
+    # Test that ListedColormap with many colors becomes ContinuousColormap
+    original_thresh = config.rc["cmap.listedthresh"]
+    try:
+        config.rc["cmap.listedthresh"] = 128
+        pcolors._cmap_database = pcolors._init_cmap_database()  # ensure pristine state
+        cmap_raw = pcolors._cmap_database._cmaps["viridis"]
+        assert not isinstance(
+            cmap_raw, (pcolors.ContinuousColormap, pcolors.DiscreteColormap)
+        )
+
+        cmap_get = pcolors._cmap_database.get_cmap("viridis")
+        assert isinstance(cmap_get, pcolors.ContinuousColormap)
+
+        cmap_raw_after = pcolors._cmap_database._cmaps["viridis"]
+        assert isinstance(cmap_raw_after, pcolors.ContinuousColormap)
+
+        # Test that ListedColormap with few colors becomes DiscreteColormap
+        config.rc["cmap.listedthresh"] = 512
+        pcolors._cmap_database = pcolors._init_cmap_database()
+        cmap_get = pcolors._cmap_database.get_cmap("viridis")
+        assert isinstance(cmap_get, pcolors.DiscreteColormap)
+    finally:
+        config.rc["cmap.listedthresh"] = original_thresh
 
 
 def test_case_insensitivity():