def setUp(self):
self.rbt = LeftLeanRedBlackTree()
self.rbt.put("E", 10)
self.rbt.put("A", 6)
self.rbt.put("R", 2)
self.rbt.put("C", 13)
self.rbt.put("H", 1)
self.rbt.put("X", -3)
self.rbt.put("M", 0)
self.rbt.put("P", 17)
self.rbt.put("L", 5)
class LeftLeanRedBlackTreeTest(TestCase):
def setUp(self):
self.rbt = LeftLeanRedBlackTree()
self.rbt.put("E", 10)
self.rbt.put("A", 6)
self.rbt.put("R", 2)
self.rbt.put("C", 13)
self.rbt.put("H", 1)
self.rbt.put("X", -3)
self.rbt.put("M", 0)
self.rbt.put("P", 17)
self.rbt.put("L", 5)
def test_get(self):
self.assertEqual(self.rbt.get("C"), 13)
self.assertEqual(self.rbt.get("M"), 0)
self.assertEqual(self.rbt.get("Z"), None)
def test_size(self):
self.assertEqual(self.rbt.size(), 9)
self.rbt.put("Z", 10)
self.assertEqual(self.rbt.size(), 10)
def test_min(self):
self.assertEqual(self.rbt.min(), "A")
def test_delete_min(self):
self.rbt.delete_min()
self.assertEqual(self.rbt.min(), "C")
self.rbt.delete_min()
self.assertEqual(self.rbt.min(), "E")
def test_delete_max(self):
self.rbt.delete_max()
self.assertEqual(self.rbt.max(), "C")
self.rbt.delete_max()
self.assertEqual(self.rbt.max(), "E")
def test_delete(self):
self.assertEqual(self.rbt.get("C"), 13)
self.rbt.delete("C")
self.assertNotEqual(self.rbt.get("C"), 13)
self.assertEqual(self.rbt.get("C"), None)
