mirror of
https://github.com/pyecharts/pyecharts.git
synced 2025-12-08 20:59:23 +00:00
123 lines
3.7 KiB
Python
123 lines
3.7 KiB
Python
# coding=utf-8
|
|
from __future__ import unicode_literals
|
|
|
|
import json
|
|
from test.utils import get_default_rendering_file_content
|
|
|
|
from pyecharts import Map
|
|
|
|
|
|
def test_map_show_label():
|
|
# show label
|
|
value = [155, 10, 66, 78]
|
|
attr = ["福建", "山东", "北京", "上海"]
|
|
map = Map("全国地图示例", width=1200, height=600)
|
|
map.add("", attr, value, maptype="china", is_label_show=True)
|
|
map.render()
|
|
|
|
|
|
def test_map_with_custom_name_map():
|
|
# show label
|
|
value = [155, 10, 66, 78]
|
|
attr = ["福建", "山东", "北京", "上海"]
|
|
map = Map("全国地图示例", width=1200, height=600)
|
|
map.add(
|
|
"",
|
|
attr,
|
|
value,
|
|
maptype="china",
|
|
is_label_show=True,
|
|
name_map={"test": "--magic--"},
|
|
)
|
|
map.render()
|
|
content = get_default_rendering_file_content()
|
|
assert "--magic--" in content
|
|
assert "nameMap" in content
|
|
|
|
|
|
def test_map_combine_with_visualmap():
|
|
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"
|
|
)
|
|
map.render()
|
|
|
|
|
|
def test_echarts_position_in_render_html():
|
|
value = [20, 190, 253, 77, 65]
|
|
attr = ["汕头市", "汕尾市", "揭阳市", "阳江市", "肇庆市"]
|
|
map = Map("广东地图示例", width=1200, height=600)
|
|
map.add(
|
|
"",
|
|
attr,
|
|
value,
|
|
maptype="广东",
|
|
is_map_symbol_show=False,
|
|
is_visualmap=True,
|
|
visual_text_color="#000",
|
|
)
|
|
map.render()
|
|
content = get_default_rendering_file_content()
|
|
echarts_position = content.find("exports.options")
|
|
guangdong_position = content.find(json.dumps("广东"))
|
|
assert echarts_position < guangdong_position
|
|
assert '"showLegendSymbol": false' in content
|
|
|
|
|
|
def test_world_map():
|
|
value = [95.1, 23.2, 43.3, 66.4, 88.5, 0.1]
|
|
attr = ["China", "Canada", "Brazil", "Russia", "United States", "Unknown Country"]
|
|
map = Map("世界地图示例", width=1200, height=600)
|
|
map.add(
|
|
"", attr, value, maptype="world", is_visualmap=True, visual_text_color="#000"
|
|
)
|
|
map.render()
|
|
|
|
actual_content = get_default_rendering_file_content()
|
|
# test register map
|
|
assert 'registerMap("world",' in actual_content
|
|
# test non-existent country
|
|
assert "Russia" in actual_content
|
|
assert "Unknown Country', " not in actual_content
|
|
|
|
|
|
def test_china_map():
|
|
value = [155, 10, 66, 78]
|
|
attr = ["福建", "山东", "北京", "上海"]
|
|
map = Map("全国地图示例", width=1200, height=600)
|
|
map.add("", attr, value, maptype="china")
|
|
map.render()
|
|
|
|
actual_content = get_default_rendering_file_content()
|
|
# test register map
|
|
assert 'registerMap("china",' in actual_content
|
|
# 福建省
|
|
assert "\u798f\u5efa" in actual_content
|
|
# 汕头市
|
|
assert "\u4e0a\u6d77" in actual_content
|
|
|
|
|
|
def test_map_visualmap_pieces():
|
|
value = [155, 10, 66, 78]
|
|
attr = ["福建", "山东", "北京", "上海"]
|
|
map = Map("全国地图示例", width=1200, height=600)
|
|
map.add(
|
|
"",
|
|
attr,
|
|
value,
|
|
maptype="china",
|
|
is_visualmap=True,
|
|
is_piecewise=True,
|
|
visual_text_color="#000",
|
|
visual_range_text=["", ""],
|
|
pieces=[
|
|
{"max": 160, "min": 70, "label": "高数值"},
|
|
{"max": 69, "min": 0, "label": "低数值"},
|
|
],
|
|
)
|
|
content = map._repr_html_()
|
|
assert '"max": 160' in content
|
|
assert '"min": 0' in content
|