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)