pyecharts/test/test_wordcloud.py
陈键冬 3e7e2bfc77 Lias (#722)
* Add: 新增 lias 工具格式化 imports

* Update: 细节修正

* Fix: fix #721

* Update: var reused

* Format: 格式化导包顺序

* Update: modules import order: python builtin imports on top, 3rd parties libraries in the middle, package's own libraries in the bottom
2018-09-01 16:07:00 +08:00

63 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_()