test/test_base.py · 朱烜宇/pyecharts

代码拉取完成,页面将自动刷新

richgiteeai

Watch

不关注关注所有动态仅关注版本发行动态关注但不提醒动态

1 Star 0 Fork 0

朱烜宇/pyecharts

代码 Issues 0 Pull Requests 0 Wiki 统计流水线

服务

加入 Gitee

与超过 1400万开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)

免费加入

已有帐号? 立即登录

文件

master

分支 (4)

标签 (24)

管理

master

dev

scatter_extra_name

0.3.x

v0.5.6

v0.5.5

v0.5.4

v0.5.3

v0.5.2

v0.5.1

v0.5.0

v0.4.1

v0.4.0

v0.3.3

v0.3.2

v0.3.1

v0.3.0

v0.2.7

v0.2.6

v0.2.4

v0.2.3

v0.2.2

v0.2.1

v0.2.0

克隆/下载

HTTPS SSH SVN SVN+SSH 下载ZIP

提示

下载代码请复制以下命令到终端执行

为确保你提交的代码身份被 Gitee 正确识别,请执行以下命令完成配置

git config --global user.name userName 
git config --global user.email userEmail

初次使用 SSH 协议进行代码克隆、推送等操作时,需按下述提示完成 SSH 配置

1 生成 RSA 密钥

2 获取 RSA 公钥内容,并配置到 SSH公钥中

在 Gitee 上使用 SVN,请访问使用指南

使用 HTTPS 协议时,命令行会出现如下账号密码验证步骤。基于安全考虑,Gitee 建议配置并使用私人令牌替代登录密码进行克隆、推送等操作

Username for 'https://gitee.com': userName

Password for 'https://userName@gitee.com': # 私人令牌

分支 4

标签 24

pyecharts

test

test_base.py

test_base.py 5.20 KB

# 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" not 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])

一键复制编辑原始数据按行查看历史

jaska 提交于 2018年05月10日 20:57 +08:00 . True theme (#546)

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202

# 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" not 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])