def test_treemap2(self):
t = TreeMap()
t.put(4, 'four')
t.put(2, 'two')
t.put(3, 'three')
t.put(1, 'one')
t.put(7, 'seven')
t.put(5, 'five')
t.put(6, 'six')
self.assertEqual(t.tree_height(), 3)
def test_treemap9(self):
t = TreeMap()
t.put(4, 'four')
t.put(2, 'two')
t.put(3, 'three')
t.put(1, 'one')
t.put(7, 'seven')
t.put(5, 'five')
t.put(6, 'six')
t.delete(6)
self.assertEqual(t.tree_height(), 2)
self.assertEqual(t.find_max(), (7,'seven'))
self.assertEqual(t.find_min(), (1,'one'))
t.delete(7)
self.assertEqual(t.tree_height(), 2)
self.assertEqual(t.find_max(), (5,'five'))
t.delete(4)
self.assertEqual(t.tree_height(), 2)
self.assertEqual(t.find_max(), (5,'five'))
self.assertEqual(t.tree.key, 5)