def test_topn_value(self):
f = Freq()
f.add(['apple', 'not', 'orange', 'juice'])
f.add(['apple', 'and', 'cinnamon', 'pie'])
returned_array = f.topn(1)
expected_array = [{'token': 'apple', 'freq': 2}]
self.assertEqual(returned_array, expected_array)
def test_topn_value(self):
f = Freq()
f.add(['apple', 'not', 'orange', 'juice'])
f.add(['apple', 'and', 'cinnamon', 'pie'])
returned_array = f.topn(1)
expected_array = [
{'token': 'apple', 'freq': 2}
]
self.assertEqual(returned_array, expected_array)
def test_topn_len(self):
f = Freq()
f.add(['apple', 'not', 'orange', 'juice'])
f.add(['apple', 'and', 'cinnamon', 'pie'])
returned_array = f.topn(5)
self.assertEqual(len(returned_array), 5)
def test_topn_len(self):
f = Freq()
f.add(['apple', 'not', 'orange', 'juice'])
f.add(['apple', 'and', 'cinnamon', 'pie'])
returned_array = f.topn(5)
self.assertEqual(len(returned_array), 5)