mirror of
https://github.com/pyecharts/pyecharts.git
synced 2025-12-08 20:59:23 +00:00
efad423d90
* 🐛 keep the empty dictionary though it looks useless but charts needs them. fix #555 * 👕 keep empty array too. update 📖 * 🎨 minor tweaks on the source code * Update: pink test folder * Update: changelog update
115 lines
3.0 KiB
Python
115 lines
3.0 KiB
Python
# coding=utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import codecs
|
|
|
|
from nose.tools import eq_
|
|
|
|
from pyecharts.utils import (
|
|
write_utf8_html_file,
|
|
get_resource_dir,
|
|
merge_js_dependencies,
|
|
)
|
|
from pyecharts.utils.lazy import LazyObject
|
|
from pyecharts.utils import remove_key_with_none_value
|
|
|
|
|
|
def test_get_resource_dir():
|
|
path = get_resource_dir("templates")
|
|
expected = os.path.join(os.getcwd(), "..", "pyecharts", "templates")
|
|
eq_(path, os.path.abspath(expected))
|
|
|
|
|
|
def test_write_utf8_html_file():
|
|
content = "柱状图数据堆叠示例"
|
|
file_name = "test.html"
|
|
write_utf8_html_file(file_name, content)
|
|
with codecs.open(file_name, "r", "utf-8") as f:
|
|
actual_content = f.read()
|
|
eq_(content, actual_content)
|
|
|
|
|
|
class MockChart(object):
|
|
"""
|
|
A mock class for a Chart and Page
|
|
"""
|
|
|
|
def __init__(self, js_dependencies):
|
|
self.js_dependencies = js_dependencies
|
|
|
|
|
|
def test_merge_js_dependencies_with_one_chart():
|
|
# Prepare some kinds of charts or page.
|
|
|
|
base_chart = MockChart(["echarts"])
|
|
|
|
# One chart or page
|
|
eq_(["echarts"], merge_js_dependencies(base_chart))
|
|
# A map chart
|
|
ch1 = MockChart(["echarts", "fujian", "zhengjiang", "anhui"])
|
|
eq_(
|
|
["echarts", "fujian", "zhengjiang", "anhui"],
|
|
merge_js_dependencies(ch1),
|
|
)
|
|
|
|
|
|
def test_merge_js_dependencies_with_multiple_charts():
|
|
base_chart = MockChart(["echarts"])
|
|
map_chart = MockChart(["echarts", "fujian"])
|
|
three_d_chart = MockChart(["echarts", "echartsgl"])
|
|
# Multiple charts
|
|
eq_(["echarts", "fujian"], merge_js_dependencies(base_chart, map_chart))
|
|
eq_(
|
|
["echarts", "echartsgl", "fujian"],
|
|
merge_js_dependencies(base_chart, map_chart, three_d_chart),
|
|
)
|
|
|
|
|
|
def test_merge_js_dependencies_with_mixed_chart_and_string():
|
|
map_chart = MockChart(["echarts", "fujian"])
|
|
|
|
eq_(["echarts", "zhejiang"], merge_js_dependencies("echarts", "zhejiang"))
|
|
eq_(
|
|
["echarts", "zhejiang"], merge_js_dependencies(["echarts", "zhejiang"])
|
|
)
|
|
eq_(["echarts", "fujian"], merge_js_dependencies("echarts", map_chart))
|
|
|
|
|
|
class MockPoint(object):
|
|
|
|
def __init__(self, x, y):
|
|
self.x = x
|
|
self.y = y
|
|
|
|
|
|
def create_point():
|
|
return MockPoint(1, 2)
|
|
|
|
|
|
def test_lazy_object():
|
|
p = LazyObject(create_point)
|
|
assert isinstance(p, LazyObject)
|
|
p.x = 3
|
|
assert isinstance(p, MockPoint)
|
|
|
|
|
|
def test_remove_key_with_none_value():
|
|
fixture = {
|
|
"a": 1,
|
|
"b": None,
|
|
"nested": {"ac": 1, "bc": None, "nested": {"a": 1, "b": None}},
|
|
"array": [
|
|
1,
|
|
{"nested": {"ac": 1, "bc": None, "nested": {"a": 1, "b": None}}},
|
|
{"normal": 1, "empty_string": ""},
|
|
],
|
|
}
|
|
actual_result = remove_key_with_none_value(fixture)
|
|
expected = {
|
|
"a": 1,
|
|
"array": [1, {"nested": {"ac": 1, "nested": {"a": 1}}}, {"normal": 1}],
|
|
"nested": {"ac": 1, "nested": {"a": 1}},
|
|
}
|
|
eq_(actual_result, expected)
|