pyecharts/test/test_line3d.py
2024-06-07 18:22:28 +08:00

39 lines
1.3 KiB
Python

import math
import unittest
from unittest.mock import patch
from pyecharts import options as opts
from pyecharts.charts import Line3D
from pyecharts.faker import Faker
class TestLine3DChart(unittest.TestCase):
@patch("pyecharts.render.engine.write_utf8_html_file")
def test_line3d_base(self, fake_writer):
data = []
for t in range(0, 25000):
_t = t / 1000
x = (1 + 0.25 * math.cos(75 * _t)) * math.cos(_t)
y = (1 + 0.25 * math.cos(75 * _t)) * math.sin(_t)
z = _t + 2.0 * math.sin(75 * _t)
data.append([x, y, z])
c = (
Line3D()
.add(
"",
data,
xaxis3d_opts=opts.Axis3DOpts(Faker.clock, type_="value"),
yaxis3d_opts=opts.Axis3DOpts(Faker.week_en, type_="value"),
grid3d_opts=opts.Grid3DOpts(width=100, height=100, depth=100),
)
.set_global_opts(
visualmap_opts=opts.VisualMapOpts(
max_=30, min_=0, range_color=Faker.visual_color
)
)
)
c.render()
_, content = fake_writer.call_args[0]
self.assertEqual(c.theme, "white")
self.assertEqual(c.renderer, "canvas")