def test_get_colors():
from bokeh.palettes import Blues5, Blues256, Viridis
from dask.diagnostics.profile_visualize import get_colors
funcs = list(range(11))
cmap = get_colors("Blues", funcs)
assert set(cmap) < set(Blues256)
assert len(set(cmap)) == 11
funcs = list(range(5))
cmap = get_colors("Blues", funcs)
lk = dict(zip(funcs, Blues5))
assert cmap == [lk[i] for i in funcs]
funcs = [0, 1, 0, 1, 0, 1]
cmap = get_colors("BrBG", funcs)
assert len(set(cmap)) == 2
funcs = list(range(100))
cmap = get_colors("Viridis", funcs)
assert len(set(cmap)) == 100
funcs = list(range(300))
cmap = get_colors("Viridis", funcs)
assert len(set(cmap)) == len(set(Viridis[256]))
def test_get_colors():
from dask.diagnostics.profile_visualize import get_colors
from bokeh.palettes import Blues9, Blues5, Viridis
from itertools import cycle
funcs = list(range(11))
cmap = get_colors('Blues', funcs)
lk = dict(zip(funcs, cycle(Blues9)))
assert cmap == [lk[i] for i in funcs]
funcs = list(range(5))
cmap = get_colors('Blues', funcs)
lk = dict(zip(funcs, Blues5))
assert cmap == [lk[i] for i in funcs]
funcs = [0, 1, 0, 1, 0, 1]
cmap = get_colors('BrBG', funcs)
assert len(set(cmap)) == 2
funcs = list(range(100))
cmap = get_colors('Viridis', funcs)
assert len(set(cmap)) == 100
funcs = list(range(300))
cmap = get_colors('Viridis', funcs)
assert len(set(cmap)) == len(set(Viridis[256]))
def test_get_colors():
from bokeh.palettes import Blues5, Viridis
from dask.diagnostics.profile_visualize import get_colors
# 256-color palettes were added in bokeh 1.4.0
if parse_version(bokeh.__version__) >= parse_version("1.4.0"):
from bokeh.palettes import Blues256
funcs = list(range(11))
cmap = get_colors("Blues", funcs)
assert set(cmap) < set(Blues256)
assert len(set(cmap)) == 11
funcs = list(range(5))
cmap = get_colors("Blues", funcs)
lk = dict(zip(funcs, Blues5))
assert cmap == [lk[i] for i in funcs]
funcs = [0, 1, 0, 1, 0, 1]
cmap = get_colors("BrBG", funcs)
assert len(set(cmap)) == 2
funcs = list(range(100))
cmap = get_colors("Viridis", funcs)
assert len(set(cmap)) == 100
funcs = list(range(300))
cmap = get_colors("Viridis", funcs)
assert len(set(cmap)) == len(set(Viridis[256]))
def test_get_colors():
from dask.diagnostics.profile_visualize import get_colors
from bokeh.palettes import Blues9, Blues5, Viridis
from itertools import cycle
funcs = list(range(11))
cmap = get_colors('Blues', funcs)
lk = dict(zip(funcs, cycle(Blues9)))
assert cmap == [lk[i] for i in funcs]
funcs = list(range(5))
cmap = get_colors('Blues', funcs)
lk = dict(zip(funcs, Blues5))
assert cmap == [lk[i] for i in funcs]
funcs = [0, 1, 0, 1, 0, 1]
cmap = get_colors('BrBG', funcs)
assert len(set(cmap)) == 2
funcs = list(range(100))
cmap = get_colors('Viridis', funcs)
assert len(set(cmap)) == 100
funcs = list(range(300))
cmap = get_colors('Viridis', funcs)
assert len(set(cmap)) == len(set(Viridis[256]))
def test_get_colors():
from dask.diagnostics.profile_visualize import get_colors
from bokeh.palettes import Blues9, Blues5, BrBG3
from itertools import cycle
funcs = list(range(11))
cmap = get_colors('Blues', funcs)
lk = dict(zip(funcs, cycle(Blues9)))
assert cmap == [lk[i] for i in funcs]
funcs = list(range(5))
cmap = get_colors('Blues', funcs)
lk = dict(zip(funcs, Blues5))
assert cmap == [lk[i] for i in funcs]
funcs = [0, 1, 0, 1, 0, 1]
cmap = get_colors('BrBG', funcs)
lk = dict(zip([0, 1], BrBG3))
assert cmap == [lk[i] for i in funcs]
def test_get_colors():
from dask.diagnostics.profile_visualize import get_colors
from bokeh.palettes import Blues9, Blues5, BrBG3
from itertools import cycle
funcs = list(range(11))
cmap = get_colors('Blues', funcs)
lk = dict(zip(funcs, cycle(Blues9)))
assert cmap == [lk[i] for i in funcs]
funcs = list(range(5))
cmap = get_colors('Blues', funcs)
lk = dict(zip(funcs, Blues5))
assert cmap == [lk[i] for i in funcs]
funcs = [0, 1, 0, 1, 0, 1]
cmap = get_colors('BrBG', funcs)
lk = dict(zip([0, 1], BrBG3))
assert cmap == [lk[i] for i in funcs]
