def test_legend_value_colors(): """Test the color_range, value_colors, and segment_colors property.""" legend = Legend(range(5), LegendParameters(segment_count=10)) for i, color in enumerate(legend.color_range): assert color == ColorRange()[i] assert legend.color_range.domain == ColorRange(domain=(0, 4)).domain assert len(legend.values) == 5 assert len(legend.value_colors) == 5 assert legend.value_colors[0] == Colorset.original()[0] assert legend.segment_colors == Colorset.original()
def test_color_range_to_from_dict(): """Test the to/from dict methods.""" color_range = ColorRange( [Color(0, 100, 0), Color(100, 200, 100)], [0, 1000]) color_range_dict = color_range.to_dict() new_color = ColorRange.from_dict(color_range_dict) assert new_color.to_dict() == color_range_dict
def test_color_range_continuous(): """Test color range objects with continuous colors.""" color_range = ColorRange( [Color(0, 100, 0), Color(100, 200, 100)], [0, 1000]) assert color_range.color(-100) == Color(0, 100, 0) assert color_range.color(0) == Color(0, 100, 0) assert color_range.color(500) == Color(50, 150, 50) assert color_range.color(250) == Color(25, 125, 25) assert color_range.color(1000) == Color(100, 200, 100) assert color_range.color(1100) == Color(100, 200, 100)
def test_init_color_range(): """Test the initialization of color range objects.""" color_range = ColorRange( [Color(75, 107, 169), Color(245, 239, 103), Color(234, 38, 0)]) str(color_range) # Test the color representation assert len(color_range) == 3 assert isinstance(color_range.colors, tuple) assert isinstance(color_range.domain, tuple) assert color_range[0] == Color(75, 107, 169)
def test_color_range_discontinuous(): """Test color range objects with discontinuous colors.""" color_range = ColorRange(continuous_colors=False) color_range.domain = [100, 2000] color_range.colors = [ Color(75, 107, 169), Color(245, 239, 103), Color(234, 38, 0) ] assert color_range.color(99) == Color(75, 107, 169) assert color_range.color(100) == Color(245, 239, 103) assert color_range.color(2000) == Color(245, 239, 103) assert color_range.color(2001) == Color(234, 38, 0)