mirror of
https://github.com/pyecharts/pyecharts.git
synced 2026-02-01 17:46:01 +00:00
c3909fee90
* #438 修复 Timeline 图累计多个 Bar 图会导致条形宽度压缩的 bug 以及不能正常显示 Tooltip 组件的 bug * 文档细节修正 * 测试修复以及版本号更新
176 lines
7.1 KiB
Python
176 lines
7.1 KiB
Python
#!/usr/bin/env python
|
|
# coding=utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
from random import randint
|
|
|
|
from pyecharts import Bar, Pie, Line, Overlap, Timeline, Style, Map
|
|
from test.constants import CLOTHES
|
|
|
|
|
|
def test_timeline_bar():
|
|
bar_1 = Bar("2012 年销量", "数据纯属虚构")
|
|
bar_1.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_1.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_1.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_1.add("冬季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
|
|
bar_2 = Bar("2013 年销量", "数据纯属虚构")
|
|
bar_2.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_2.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_2.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_2.add("冬季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
|
|
bar_3 = Bar("2014 年销量", "数据纯属虚构")
|
|
bar_3.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_3.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_3.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_3.add("冬季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
|
|
bar_4 = Bar("2015 年销量", "数据纯属虚构")
|
|
bar_4.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_4.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_4.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_4.add("冬季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
|
|
bar_5 = Bar("2016 年销量", "数据纯属虚构")
|
|
bar_5.add("春季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_5.add("夏季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_5.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)])
|
|
bar_5.add("冬季", 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 年')
|
|
timeline.add(bar_3, '2014 年')
|
|
timeline.add(bar_4, '2015 年')
|
|
timeline.add(bar_5, '2016 年')
|
|
assert len(timeline.options.get("baseOption").get("series")) == 0
|
|
timeline.render()
|
|
|
|
|
|
def test_timeline_pie():
|
|
style = Style()
|
|
pie_style = style.add(
|
|
is_label_show=True,
|
|
radius=[30, 55],
|
|
rosetype="radius"
|
|
)
|
|
pie_1 = Pie("2012 年销量比例", "数据纯属虚构")
|
|
pie_1.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)], **pie_style)
|
|
|
|
pie_2 = Pie("2013 年销量比例", "数据纯属虚构")
|
|
pie_2.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)], **pie_style)
|
|
|
|
pie_3 = Pie("2014 年销量比例", "数据纯属虚构")
|
|
pie_3.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)], **pie_style)
|
|
|
|
pie_4 = Pie("2015 年销量比例", "数据纯属虚构")
|
|
pie_4.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)], **pie_style)
|
|
|
|
pie_5 = Pie("2016 年销量比例", "数据纯属虚构")
|
|
pie_5.add("秋季", CLOTHES, [randint(10, 100) for _ in range(6)], **pie_style)
|
|
|
|
timeline = Timeline(is_auto_play=True, timeline_bottom=0,
|
|
width=1200, height=600)
|
|
timeline.add(pie_1, '2012 年')
|
|
timeline.add(pie_2, '2013 年')
|
|
timeline.add(pie_3, '2014 年')
|
|
timeline.add(pie_4, '2015 年')
|
|
timeline.add(pie_5, '2016 年')
|
|
assert len(timeline.options.get("baseOption").get("series")) == 0
|
|
timeline.render()
|
|
|
|
|
|
def test_timeline_bar_line():
|
|
attr = ["{}月".format(i) for i in range(1, 7)]
|
|
bar = Bar("1 月份数据", "数据纯属虚构")
|
|
bar.add("bar", attr, [randint(10, 50) for _ in range(6)])
|
|
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)
|
|
|
|
bar_2 = Bar("3 月份数据", "数据纯属虚构")
|
|
bar_2.add("bar", attr, [randint(10, 50) for _ in range(6)])
|
|
line_2 = Line()
|
|
line_2.add("line", attr, [randint(50, 80) for _ in range(6)])
|
|
overlap_2 = Overlap()
|
|
overlap_2.add(bar_2)
|
|
overlap_2.add(line_2)
|
|
|
|
bar_3 = Bar("4 月份数据", "数据纯属虚构")
|
|
bar_3.add("bar", attr, [randint(10, 50) for _ in range(6)])
|
|
line_3 = Line()
|
|
line_3.add("line", attr, [randint(50, 80) for _ in range(6)])
|
|
overlap_3 = Overlap()
|
|
overlap_3.add(bar_3)
|
|
overlap_3.add(line_3)
|
|
|
|
bar_4 = Bar("5 月份数据", "数据纯属虚构")
|
|
bar_4.add("bar", attr, [randint(10, 50) for _ in range(6)])
|
|
line_4 = Line()
|
|
line_4.add("line", attr, [randint(50, 80) for _ in range(6)])
|
|
overlap_4 = Overlap()
|
|
overlap_4.add(bar_4)
|
|
overlap_4.add(line_4)
|
|
|
|
timeline = Timeline(timeline_bottom=0)
|
|
timeline.add(overlap_0, '1 月')
|
|
timeline.add(overlap_1, '2 月')
|
|
timeline.add(overlap_2, '3 月')
|
|
timeline.add(overlap_3, '4 月')
|
|
timeline.add(overlap_4, '5 月')
|
|
timeline.render()
|
|
|
|
|
|
def test_timeline_map():
|
|
timeline = Timeline(timeline_bottom=0)
|
|
value = [155, 10, 66, 78, 33, 80, 190, 53, 49.6]
|
|
attr = ["福建", "山东", "北京", "上海", "甘肃",
|
|
"新疆", "河南", "广西", "西藏"]
|
|
map = Map("Map 结合 VisualMap 示例", width=1200, height=600)
|
|
map.add("", attr, value, maptype='china', is_visualmap=True,
|
|
visual_text_color='#000', visual_top="30%")
|
|
timeline.add(map, "test1")
|
|
value = [155, 10, 66, 78, 33]
|
|
attr = ["福建", "山东", "北京", "上海", "甘肃"]
|
|
map = Map("Map 结合 VisualMap 示例", width=1200, height=600)
|
|
map.add("", attr, value, maptype='china', is_visualmap=True,
|
|
visual_text_color='#000', visual_top="30%")
|
|
timeline.add(map, "test2")
|
|
assert len(timeline.options.get("baseOption").get("series")) == 2
|
|
timeline.render()
|
|
|
|
|
|
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
|