def test_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
]
# 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_()
# 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_()
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_()
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_()
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_()
