mirror of
https://github.com/pyecharts/pyecharts.git
synced 2026-02-01 17:46:01 +00:00
fd16ffb66d
* 🐛 turn on split line for scatter diagram. fix #516 * 🔥 remove documentation that echos the function name * ✨ support dark theme. fix #512 * ✨ support global_theme in configure function. improves #512 * ✨ use_theme, related to #512 * ✨ support 5 more external themes which come from echarts-themes-pypkg and echarts-themes-js, #512 * 📰 add missing test fixtures * 💚 make unit test pass * 📚 document themes usage * 📚 add cross reference * 📚 update change log and pump version * Update: update some docs * Update: typo fixed * Update: prepare docs update * Update: 鉴于现在 md 做 long description 效果不佳，转化为 rst 又出现问题，所以暂时取消 * Fix: fixed broken test, remove unused codecs
203 lines
5.2 KiB
Python
203 lines
5.2 KiB
Python
# coding=utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
|
|
import pandas as pd
|
|
import numpy as np
|
|
|
|
from nose.tools import eq_, raises
|
|
from mock import patch, MagicMock
|
|
|
|
from pyecharts import Bar, Map, jupyter_image
|
|
import pyecharts.exceptions as exceptions
|
|
from test.constants import CLOTHES
|
|
from test.utils import get_default_rendering_file_content
|
|
|
|
TITLE = "柱状图数据堆叠示例"
|
|
|
|
|
|
def create_a_bar(title, renderer='canvas'):
|
|
v1 = [5, 20, 36, 10, 75, 90]
|
|
v2 = [10, 25, 8, 60, 20, 80]
|
|
bar = Bar(title, renderer=renderer)
|
|
bar.add("商家A", CLOTHES, v1, is_stack=True)
|
|
bar.add("商家B", CLOTHES, v2, is_stack=True)
|
|
return bar
|
|
|
|
|
|
def test_theme_option():
|
|
bar = create_a_bar(TITLE, renderer='svg')
|
|
bar.use_theme('dark')
|
|
html = bar.render_embed()
|
|
assert "'dark'" in html
|
|
|
|
|
|
@raises(exceptions.InvalidTheme)
|
|
def test_invalid_theme_option():
|
|
bar = create_a_bar(TITLE, renderer='svg')
|
|
bar.use_theme('brilliant')
|
|
|
|
|
|
def test_svg_option():
|
|
bar = create_a_bar(TITLE, renderer='svg')
|
|
html = bar.render_embed()
|
|
assert "{renderer: 'svg'}" in html
|
|
|
|
|
|
def test_svg_option_in_note_book():
|
|
bar = create_a_bar(TITLE, renderer='svg')
|
|
html = bar._repr_html_()
|
|
assert "{renderer: 'svg'}" in html
|
|
|
|
|
|
def test_canvas_option():
|
|
bar = create_a_bar(TITLE)
|
|
html = bar.render_embed()
|
|
assert "{renderer: 'canvas'}" in html
|
|
|
|
|
|
def test_canvas_option_in_notebook():
|
|
bar = create_a_bar(TITLE)
|
|
html = bar._repr_html_()
|
|
assert "{renderer: 'canvas'}" in html
|
|
|
|
|
|
def test_embed_option():
|
|
bar = create_a_bar(TITLE)
|
|
html = bar.render_embed()
|
|
json_encoded_title = json.dumps(TITLE)
|
|
assert json_encoded_title in html
|
|
assert "<html>" not in html
|
|
assert "<body>" not in html
|
|
|
|
|
|
def test_jupyter_repr_png():
|
|
bar = create_a_bar(TITLE)
|
|
assert bar._repr_png_() is None
|
|
|
|
|
|
def test_jupyter_repr_jpeg():
|
|
bar = create_a_bar(TITLE)
|
|
assert bar._repr_jpeg_() is None
|
|
|
|
|
|
def test_jupyter_repr_svg():
|
|
bar = create_a_bar(TITLE)
|
|
assert bar._repr_svg_() is None
|
|
|
|
|
|
@patch('pyecharts.engine.create_default_environment')
|
|
@patch('os.unlink')
|
|
def test_render_as_svg(fake_unlink, fake_factory):
|
|
fake_env = MagicMock(
|
|
render_chart_to_file=MagicMock(return_value="fake svg")
|
|
)
|
|
fake_factory.return_value = fake_env
|
|
with jupyter_image('svg'):
|
|
bar = create_a_bar('test', renderer='svg')
|
|
svg_content = bar._repr_svg_()
|
|
fake_unlink.assert_called()
|
|
eq_(svg_content, 'fake svg')
|
|
|
|
|
|
@raises(exceptions.InvalidConfiguration)
|
|
def test_render_as_svg_with_wrong_configuration():
|
|
with jupyter_image('svg'):
|
|
bar = create_a_bar('test')
|
|
bar._repr_svg_()
|
|
|
|
|
|
@raises(exceptions.InvalidConfiguration)
|
|
def test_render_as_png_with_wrong_configuration():
|
|
with jupyter_image('png'):
|
|
bar = create_a_bar('test', renderer='svg')
|
|
bar._repr_png_()
|
|
|
|
|
|
def test_base_get_js_dependencies():
|
|
bar = create_a_bar(TITLE)
|
|
dependencies = bar.get_js_dependencies()
|
|
expected = ['echarts.min']
|
|
eq_(dependencies, expected)
|
|
|
|
|
|
def test_numpy_array():
|
|
v1 = np.array([5, 20, 36, 10, 75, 90])
|
|
bar = Bar(TITLE)
|
|
bar.add("商家A", CLOTHES, v1, is_stack=True)
|
|
html = bar.render_embed()
|
|
json_encoded_title = json.dumps(TITLE)
|
|
assert json_encoded_title in html
|
|
|
|
|
|
def test_pandas_dataframe():
|
|
title = 'bar chart'
|
|
index = pd.date_range('3/8/2017', periods=6, freq='M')
|
|
df1 = pd.DataFrame(np.random.randn(6), index=index)
|
|
df2 = pd.DataFrame(np.random.randn(6), index=index)
|
|
|
|
dtvalue1 = [i[0] for i in df1.values]
|
|
dtvalue2 = [i[0] for i in df2.values]
|
|
_index = [i for i in df1.index.format()]
|
|
|
|
bar = Bar(title, 'Profit and loss situation')
|
|
bar.add('profit', _index, dtvalue1)
|
|
bar.add('loss', _index, dtvalue2)
|
|
html = bar.render_embed()
|
|
assert title in html
|
|
|
|
|
|
def test_echarts_position_in_render_html():
|
|
value = [20, 190, 253, 77, 65]
|
|
attr = ['汕头市', '汕尾市', '揭阳市', '阳江市', '肇庆市']
|
|
map = Map("广东地图示例", width=1200, height=600, page_title=TITLE)
|
|
map.add(
|
|
"",
|
|
attr,
|
|
value,
|
|
maptype='广东',
|
|
is_visualmap=True,
|
|
visual_text_color='#000',
|
|
)
|
|
map.render()
|
|
actual_content = get_default_rendering_file_content()
|
|
assert TITLE in actual_content
|
|
|
|
|
|
def test_show_config():
|
|
stdout_ = sys.stdout
|
|
captured_stdout = 'stdout.txt'
|
|
try:
|
|
with open(captured_stdout, 'w') as f:
|
|
sys.stdout = f
|
|
bar = create_a_bar("new")
|
|
bar.print_echarts_options()
|
|
except Exception as e:
|
|
# whatever happens, continue and restore stdout
|
|
print(e)
|
|
sys.stdout = stdout_
|
|
with open(captured_stdout, 'r') as f:
|
|
content = f.read()
|
|
assert 'None' not in content
|
|
assert 'null' in content
|
|
assert 'false' in content
|
|
assert 'False' not in content
|
|
os.unlink(captured_stdout)
|
|
|
|
|
|
def test_base_cast_records():
|
|
records = [{"key": 1}, {"value": 2}]
|
|
keys, values = Bar.cast(records)
|
|
eq_(keys, ["key", "value"])
|
|
eq_(values, [1, 2])
|
|
|
|
|
|
def test_base_cast_dict():
|
|
adict = {"key": 1, "value": 2}
|
|
keys, values = Bar.cast(adict)
|
|
eq_(keys, ["key", "value"])
|
|
eq_(values, [1, 2])
|