pyecharts/test/test_wordcloud.py
陈键冬 8f16369400 Charts docs update and pink everything (#536)
* Update: docs update

* Update: 统一注释文档风格

* Update: pink echarts folder

* Update: pink charts folder

* Update: pink custom folder

* Update: pink pyecharts and dataset folder

* Update: pink test folder

* Update: pink utils folder

* Update: micro docs update for theme

* Fix: fixed broken test
2018-05-04 22:37:27 +08:00

64 lines
1.1 KiB
Python

#!/usr/bin/env python
# coding=utf-8
from pyecharts import WordCloud
name = [
"Sam S Club",
"Macys",
"Amy Schumer",
"Jurassic World",
"Charter Communications",
"Chick Fil A",
"Planet Fitness",
"Pitch Perfect",
"Express",
"Home",
"Johnny Depp",
"Lena Dunham",
"Lewis Hamilton",
"KXAN",
"Mary Ellen Mark",
"Farrah Abraham",
"Rita Ora",
"Serena Williams",
"NCAA baseball tournament",
"Point Break",
]
value = [
10000,
6181,
4386,
4055,
2467,
2244,
1898,
1484,
1112,
965,
847,
582,
555,
550,
462,
366,
360,
282,
273,
265,
]
def test_wordcloud_default():
wordcloud = WordCloud(width=1300, height=620)
wordcloud.add("", name, value, word_size_range=[30, 100], rotate_step=66)
assert "diamond" not in wordcloud._repr_html_()
def test_wordcloud_shape_diamond():
wordcloud = WordCloud(width=1300, height=620)
wordcloud.add("", name, value, word_size_range=[30, 100], shape="diamond")
assert "diamond" in wordcloud._repr_html_()