def test_parsimonious_wordcloud():
docs = map(lambda s: s.split(),
["an apple a day keeps the doctor away",
"orange is the new black",
"comparing apples and oranges"])
lm = parsimonious_wordcloud(docs, k=4)
assert_equal(3, len(lm))
for x in lm:
assert_equal(4, len(x))
for term, weight in x:
assert_true(isinstance(term, basestring))
assert_true(isinstance(weight, numbers.Real))
def test_wordcloud():
cloud = parsimonious_wordcloud([doc.split() for doc in DOCS])
assert_equal(len(cloud), len(DOCS))
assert_equal(len(cloud[0]), 10)
def test_wordcloud():
cloud = parsimonious_wordcloud([doc.split() for doc in DOCS])
assert_equal(len(cloud), len(DOCS))
assert_equal(len(cloud[0]), 10)
