def test_max_list(self):
ml = MaxList(3)
ml.add(5)
ml.add(2)
ml.add(1)
self.assertEqual(ml.get_list(), [5, 2, 1])
ml.add(3)
ml.add(1)
self.assertEqual(ml.get_list(), [5, 3, 2])
def test_max_item(self):
ml = MaxList(1)
ml.add(5)
ml.add(2)
ml.add(8)
ml.add(3)
self.assertEqual(ml.get_list(), [8])
def test_max_list_string(self):
ml = MaxList(4, key=lambda x: len(x))
ml.add("American Trade Deal")
ml.add("Chryseler")
ml.add("Supercalafragilisticexpialodocious")
ml.add("ravenclaw")
ml.add("Facebook Rocks!")
self.assertEqual(ml.get_list()[3], "Facebook Rocks!")
self.assertEqual(len(ml.get_list()), 4)