mirror of
https://github.com/pyecharts/pyecharts.git
synced 2025-12-08 20:59:23 +00:00
437ae7d7f6
* 🐛 fix theme color problem for themes other than dark, #544 * 🔬 test remove_key_with_none_value and whitening * 📚 update docs and pump version * Update json_serializable.py * Update __init__.py * 💚 fix unit test failures * 🔥 get rid of dev code * 💚 pass all unit tests * 👕 use python 2 to generate the fixtures * 🔥 remove empty dictionary that resulted from cleaning * 🔬 more test cases. 👕 whitening * 💚 ignore basestring please, flake8 * 🔨 code refactor test * 🔨 address review feedbacks * 💄 change single quote * Update: docs update * Update: img link update * Update: docs update
229 lines
8.1 KiB
Python
229 lines
8.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._option.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._option.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._option.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
|
|
|
|
|
|
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
|