def test_timeline_label_color(): attr = ["{}月".format(i) for i in range(1, 7)] bar = Bar("1 月份数据", "数据纯属虚构") bar.add( "bar", attr, [randint(10, 50) for _ in range(6)], label_color=["red", "#213", "black"], ) line = Line() line.add("line", attr, [randint(50, 80) for _ in range(6)]) overlap_0 = Overlap() overlap_0.add(bar) overlap_0.add(line) bar_1 = Bar("2 月份数据", "数据纯属虚构") bar_1.add("bar", attr, [randint(10, 50) for _ in range(6)]) line_1 = Line() line_1.add("line", attr, [randint(50, 80) for _ in range(6)]) overlap_1 = Overlap() overlap_1.add(bar_1) overlap_1.add(line_1) timeline = Timeline(timeline_bottom=0) timeline.add(overlap_0, "1 月") timeline.add(overlap_1, "2 月") content = timeline._repr_html_() assert '"color": [' in content assert "red" in content assert "#213" in content assert "black" in content
def test_timeline_different_legend(): bar_1 = Bar("2012 年销量", "数据纯属虚构") bar_1.add("春季a", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_1.add("夏季a", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_1.add("秋季a", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_1.add("冬季a", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_2 = Bar("2013 年销量", "数据纯属虚构") bar_2.add("春季b", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_2.add("夏季b", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_2.add("秋季b", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_2.add("冬季b", CLOTHES, [randint(10, 100) for _ in range(6)], is_legend_show=True) timeline = Timeline(is_auto_play=True, timeline_bottom=0) timeline.add(bar_1, '2012 年') timeline.add(bar_2, '2013 年') content = timeline._repr_html_() assert "\\u6625\\u5b63a" in content # 春季 a assert "\\u6625\\u5b63b" in content # 春季 b
def test_timeline_different_legend(): bar_1 = Bar("2012 年销量", "数据纯属虚构") bar_1.add("春季a", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_1.add("夏季a", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_1.add("秋季a", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_1.add("冬季a", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_2 = Bar("2013 年销量", "数据纯属虚构") bar_2.add("春季b", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_2.add("夏季b", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_2.add("秋季b", CLOTHES, [randint(10, 100) for _ in range(6)]) bar_2.add( "冬季b", CLOTHES, [randint(10, 100) for _ in range(6)], is_legend_show=True, ) timeline = Timeline(is_auto_play=True, timeline_bottom=0) timeline.add(bar_1, "2012 年") timeline.add(bar_2, "2013 年") content = timeline._repr_html_() assert "\\u6625\\u5b63a" in content # 春季 a assert "\\u6625\\u5b63b" in content # 春季 b