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]