mirror of
https://github.com/pyecharts/pyecharts.git
synced 2025-12-08 20:59:23 +00:00
331 lines
13 KiB
Python
331 lines
13 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from pyecharts import options as opts
|
|
from pyecharts.charts import Map3D
|
|
from pyecharts.faker import Faker
|
|
from pyecharts.globals import ChartType
|
|
from pyecharts.commons.utils import JsCode
|
|
|
|
|
|
class TestMap3DChart(unittest.TestCase):
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_map3d_base(self, fake_writer):
|
|
c = (
|
|
Map3D()
|
|
.add_schema()
|
|
.add(
|
|
"商家A",
|
|
[list(z) for z in zip(Faker.provinces, Faker.values())],
|
|
maptype="china",
|
|
)
|
|
)
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
self.assertIn("map3D", content)
|
|
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_map3d_schema(self, fake_writer):
|
|
c = (
|
|
Map3D()
|
|
.add_schema(
|
|
itemstyle_opts=opts.ItemStyleOpts(),
|
|
map3d_label=opts.Map3DLabelOpts(),
|
|
light_opts=opts.Map3DLightOpts(),
|
|
view_control_opts=opts.Map3DViewControlOpts(),
|
|
post_effect_opts=opts.Map3DPostEffectOpts(),
|
|
realistic_material_opts=opts.Map3DRealisticMaterialOpts(),
|
|
lambert_material_opts=opts.Map3DLambertMaterialOpts(),
|
|
color_material_opts=opts.Map3DColorMaterialOpts(),
|
|
)
|
|
.add(
|
|
series_name="商家A",
|
|
data_pair=[list(z) for z in zip(Faker.provinces, Faker.values())],
|
|
maptype="china",
|
|
type_=ChartType.LINES3D,
|
|
effect=opts.Lines3DEffectOpts(),
|
|
)
|
|
)
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
self.assertIn("itemStyle", content)
|
|
self.assertIn("label", content)
|
|
self.assertIn("light", content)
|
|
self.assertIn("viewControl", content)
|
|
self.assertIn("postEffect", content)
|
|
self.assertIn("realisticMaterial", content)
|
|
self.assertIn("lambertMaterial", content)
|
|
self.assertIn("colorMaterial", content)
|
|
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_map3d_with_bar3d(self, fake_writer):
|
|
example_data = [
|
|
("黑龙江", [127.9688, 45.368, 100]),
|
|
("内蒙古", [110.3467, 41.4899, 300]),
|
|
("吉林", [125.8154, 44.2584, 300]),
|
|
("辽宁", [123.1238, 42.1216, 300]),
|
|
("河北", [114.4995, 38.1006, 300]),
|
|
("天津", [117.4219, 39.4189, 300]),
|
|
("山西", [112.3352, 37.9413, 300]),
|
|
("陕西", [109.1162, 34.2004, 300]),
|
|
("甘肃", [103.5901, 36.3043, 300]),
|
|
("宁夏", [106.3586, 38.1775, 300]),
|
|
("青海", [101.4038, 36.8207, 300]),
|
|
("新疆", [87.9236, 43.5883, 300]),
|
|
("西藏", [91.11, 29.97, 300]),
|
|
("四川", [103.9526, 30.7617, 300]),
|
|
("重庆", [108.384366, 30.439702, 300]),
|
|
("山东", [117.1582, 36.8701, 300]),
|
|
("河南", [113.4668, 34.6234, 300]),
|
|
("江苏", [118.8062, 31.9208, 300]),
|
|
("安徽", [117.29, 32.0581, 300]),
|
|
("湖北", [114.3896, 30.6628, 300]),
|
|
("浙江", [119.5313, 29.8773, 300]),
|
|
("福建", [119.4543, 25.9222, 300]),
|
|
("江西", [116.0046, 28.6633, 300]),
|
|
("湖南", [113.0823, 28.2568, 300]),
|
|
("贵州", [106.6992, 26.7682, 300]),
|
|
("广西", [108.479, 23.1152, 300]),
|
|
("海南", [110.3893, 19.8516, 300]),
|
|
("上海", [121.4648, 31.2891, 1300]),
|
|
]
|
|
|
|
c = (
|
|
Map3D()
|
|
.add_schema(
|
|
itemstyle_opts=opts.ItemStyleOpts(
|
|
color="rgb(5,101,123)",
|
|
opacity=1,
|
|
border_width=0.8,
|
|
border_color="rgb(62,215,213)",
|
|
),
|
|
map3d_label=opts.Map3DLabelOpts(
|
|
is_show=False,
|
|
formatter=JsCode(
|
|
"function(data){return data.name + " " + data.value[2];}"
|
|
),
|
|
),
|
|
emphasis_label_opts=opts.LabelOpts(
|
|
is_show=False,
|
|
color="#fff",
|
|
font_size=10,
|
|
background_color="rgba(0,23,11,0)",
|
|
),
|
|
light_opts=opts.Map3DLightOpts(
|
|
main_color="#fff",
|
|
main_intensity=1.2,
|
|
main_shadow_quality="high",
|
|
is_main_shadow=False,
|
|
main_beta=10,
|
|
ambient_intensity=0.3,
|
|
),
|
|
)
|
|
.add(
|
|
series_name="bar3D",
|
|
data_pair=example_data,
|
|
type_=ChartType.BAR3D,
|
|
bar_size=1,
|
|
shading="lambert",
|
|
label_opts=opts.LabelOpts(
|
|
is_show=False,
|
|
formatter=JsCode(
|
|
"function(data){return data.name + ' ' + data.value[2];}"
|
|
),
|
|
),
|
|
)
|
|
.set_global_opts(title_opts=opts.TitleOpts(title="Map3D-Bar3D"))
|
|
)
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
self.assertIn("itemStyle", content)
|
|
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_map3d_with_lines3d(self, fake_writer):
|
|
example_data = [
|
|
[[119.107078, 36.70925, 1000], [116.587245, 35.415393, 1000]],
|
|
[[117.000923, 36.675807], [120.355173, 36.082982]],
|
|
[[118.047648, 36.814939], [118.66471, 37.434564]],
|
|
[[121.391382, 37.539297], [119.107078, 36.70925]],
|
|
[[116.587245, 35.415393], [122.116394, 37.509691]],
|
|
[[119.461208, 35.428588], [118.326443, 35.065282]],
|
|
[[116.307428, 37.453968], [115.469381, 35.246531]],
|
|
]
|
|
c = (
|
|
Map3D()
|
|
.add_schema(
|
|
maptype="山东",
|
|
itemstyle_opts=opts.ItemStyleOpts(
|
|
color="rgb(5,101,123)",
|
|
opacity=1,
|
|
border_width=0.8,
|
|
border_color="rgb(62,215,213)",
|
|
),
|
|
light_opts=opts.Map3DLightOpts(
|
|
main_color="#fff",
|
|
main_intensity=1.2,
|
|
is_main_shadow=False,
|
|
main_alpha=55,
|
|
main_beta=10,
|
|
ambient_intensity=0.3,
|
|
),
|
|
view_control_opts=opts.Map3DViewControlOpts(center=[-10, 0, 10]),
|
|
post_effect_opts=opts.Map3DPostEffectOpts(is_enable=False),
|
|
)
|
|
.add(
|
|
series_name="",
|
|
data_pair=example_data,
|
|
type_=ChartType.LINES3D,
|
|
effect=opts.Lines3DEffectOpts(
|
|
is_show=True,
|
|
period=4,
|
|
trail_width=3,
|
|
trail_length=0.5,
|
|
trail_color="#f00",
|
|
trail_opacity=1,
|
|
),
|
|
linestyle_opts=opts.LineStyleOpts(
|
|
is_show=False, color="#fff", opacity=0
|
|
),
|
|
)
|
|
.set_global_opts(title_opts=opts.TitleOpts(title="Map3D-Lines3D"))
|
|
)
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
self.assertIn("itemStyle", content)
|
|
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_map3d_with_line3d(self, fake_writer):
|
|
example_data = [
|
|
[[119.107078, 36.70925, 1000], [116.587245, 35.415393, 1000]],
|
|
[[117.000923, 36.675807], [120.355173, 36.082982]],
|
|
[[118.047648, 36.814939], [118.66471, 37.434564]],
|
|
[[121.391382, 37.539297], [119.107078, 36.70925]],
|
|
[[116.587245, 35.415393], [122.116394, 37.509691]],
|
|
[[119.461208, 35.428588], [118.326443, 35.065282]],
|
|
[[116.307428, 37.453968], [115.469381, 35.246531]],
|
|
]
|
|
c = (
|
|
Map3D()
|
|
.add_schema(
|
|
maptype="山东",
|
|
itemstyle_opts=opts.ItemStyleOpts(
|
|
color="rgb(5,101,123)",
|
|
opacity=1,
|
|
border_width=0.8,
|
|
border_color="rgb(62,215,213)",
|
|
),
|
|
light_opts=opts.Map3DLightOpts(
|
|
main_color="#fff",
|
|
main_intensity=1.2,
|
|
is_main_shadow=False,
|
|
main_alpha=55,
|
|
main_beta=10,
|
|
ambient_intensity=0.3,
|
|
),
|
|
view_control_opts=opts.Map3DViewControlOpts(center=[-10, 0, 10]),
|
|
post_effect_opts=opts.Map3DPostEffectOpts(is_enable=False),
|
|
)
|
|
.add(
|
|
series_name="",
|
|
data_pair=example_data,
|
|
type_=ChartType.LINE3D,
|
|
effect=opts.Lines3DEffectOpts(
|
|
is_show=True,
|
|
period=4,
|
|
trail_width=3,
|
|
trail_length=0.5,
|
|
trail_color="#f00",
|
|
trail_opacity=1,
|
|
),
|
|
linestyle_opts=opts.LineStyleOpts(
|
|
is_show=False, color="#fff", opacity=0
|
|
),
|
|
)
|
|
.set_global_opts(title_opts=opts.TitleOpts(title="Map3D-Line3D"))
|
|
)
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
self.assertIn("itemStyle", content)
|
|
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_map3d_with_scatter3d(self, fake_writer):
|
|
example_data = [
|
|
("黑龙江", [127.9688, 45.368, 100]),
|
|
("内蒙古", [110.3467, 41.4899, 100]),
|
|
("吉林", [125.8154, 44.2584, 100]),
|
|
("辽宁", [123.1238, 42.1216, 100]),
|
|
("河北", [114.4995, 38.1006, 100]),
|
|
("天津", [117.4219, 39.4189, 100]),
|
|
("山西", [112.3352, 37.9413, 100]),
|
|
("陕西", [109.1162, 34.2004, 100]),
|
|
("甘肃", [103.5901, 36.3043, 100]),
|
|
("宁夏", [106.3586, 38.1775, 100]),
|
|
("青海", [101.4038, 36.8207, 100]),
|
|
("新疆", [87.9236, 43.5883, 100]),
|
|
("西藏", [91.11, 29.97, 100]),
|
|
("四川", [103.9526, 30.7617, 100]),
|
|
("重庆", [108.384366, 30.439702, 100]),
|
|
("山东", [117.1582, 36.8701, 100]),
|
|
("河南", [113.4668, 34.6234, 100]),
|
|
("江苏", [118.8062, 31.9208, 100]),
|
|
("安徽", [117.29, 32.0581, 100]),
|
|
("湖北", [114.3896, 30.6628, 100]),
|
|
("浙江", [119.5313, 29.8773, 100]),
|
|
("福建", [119.4543, 25.9222, 100]),
|
|
("江西", [116.0046, 28.6633, 100]),
|
|
("湖南", [113.0823, 28.2568, 100]),
|
|
("贵州", [106.6992, 26.7682, 100]),
|
|
("广西", [108.479, 23.1152, 100]),
|
|
("海南", [110.3893, 19.8516, 100]),
|
|
("上海", [121.4648, 31.2891, 100]),
|
|
]
|
|
|
|
c = (
|
|
Map3D()
|
|
.add_schema(
|
|
itemstyle_opts=opts.ItemStyleOpts(
|
|
color="rgb(5,101,123)",
|
|
opacity=1,
|
|
border_width=0.8,
|
|
border_color="rgb(62,215,213)",
|
|
),
|
|
map3d_label=opts.Map3DLabelOpts(
|
|
is_show=False,
|
|
formatter=JsCode(
|
|
"function(data){return data.name + " " + data.value[2];}"
|
|
),
|
|
),
|
|
emphasis_label_opts=opts.LabelOpts(
|
|
is_show=False,
|
|
color="#fff",
|
|
font_size=10,
|
|
background_color="rgba(0,23,11,0)",
|
|
),
|
|
light_opts=opts.Map3DLightOpts(
|
|
main_color="#fff",
|
|
main_intensity=1.2,
|
|
main_shadow_quality="high",
|
|
is_main_shadow=False,
|
|
main_beta=10,
|
|
ambient_intensity=0.3,
|
|
),
|
|
)
|
|
.add(
|
|
series_name="Scatter3D",
|
|
data_pair=example_data,
|
|
type_=ChartType.SCATTER3D,
|
|
bar_size=1,
|
|
shading="lambert",
|
|
label_opts=opts.LabelOpts(
|
|
is_show=False,
|
|
formatter=JsCode(
|
|
"function(data){return data.name + ' ' + data.value[2];}"
|
|
),
|
|
),
|
|
)
|
|
.set_global_opts(title_opts=opts.TitleOpts(title="Map3D-Scatter3D"))
|
|
)
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
self.assertIn("itemStyle", content)
|