mirror of
https://github.com/pyecharts/pyecharts.git
synced 2026-01-25 17:06:27 +00:00
63 lines
1.7 KiB
Python
63 lines
1.7 KiB
Python
from unittest.mock import patch
|
|
|
|
from nose.tools import assert_equal, assert_in
|
|
|
|
from pyecharts.charts import WordCloud
|
|
from pyecharts.commons.utils import JsCode
|
|
from pyecharts.exceptions import WordCloudMaskImageException
|
|
|
|
words = [
|
|
("Sam S Club", 10000),
|
|
("Macys", 6181),
|
|
("Amy Schumer", 4386),
|
|
("Jurassic World", 4055),
|
|
("Charter Communications", 2467),
|
|
("Chick Fil A", 2244),
|
|
("Planet Fitness", 1868),
|
|
("Pitch Perfect", 1484),
|
|
]
|
|
|
|
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_wordcloud_base(fake_writer):
|
|
c = WordCloud().add("", words, word_size_range=[20, 100])
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
assert_equal(c.theme, "white")
|
|
assert_equal(c.renderer, "canvas")
|
|
|
|
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_wordcloud_shapes(fake_writer):
|
|
c = WordCloud().add("", words, word_size_range=[20, 100], shape="cardioid")
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
assert_equal(c.theme, "white")
|
|
assert_equal(c.renderer, "canvas")
|
|
|
|
|
|
def test_wordcloud_error_url():
|
|
try:
|
|
c = WordCloud().add(
|
|
"", words, word_size_range=[20, 100], mask_image="error images_url"
|
|
)
|
|
c.render()
|
|
except WordCloudMaskImageException as err:
|
|
assert_equal(type(err), WordCloudMaskImageException)
|
|
assert err.__str__() != ""
|
|
|
|
|
|
@patch("pyecharts.render.engine.write_utf8_html_file")
|
|
def test_wordcloud_mask_image(fake_writer):
|
|
c = WordCloud().add(
|
|
"",
|
|
words,
|
|
word_size_range=[20, 100],
|
|
shape="cardioid",
|
|
mask_image="test/fixtures/img.png",
|
|
)
|
|
c.render()
|
|
_, content = fake_writer.call_args[0]
|
|
assert_equal(c.theme, "white")
|
|
assert_equal(c.renderer, "canvas")
|