mirror of
https://github.com/pyecharts/pyecharts.git
synced 2026-02-01 17:46:01 +00:00
26840eb9d3
* 🔨 code refactoring using https://github.com/kennethreitz/white * 🔨 code refactoring the test code using https://github.com/kennethreitz/white
64 lines
1.1 KiB
Python
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_()
|