def test_avl_is_balancing_correctly(self): """ Test that it is balancing correctly """ avl = AVLTree() point = VirtualPoint("10.128.20.1", 20) avl.insert(20, point) point = VirtualPoint("10.128.20.2", 4) avl.insert(4, point) point = VirtualPoint("10.128.20.3", 3) avl.insert(3, point) point = VirtualPoint("10.128.20.4", 9) avl.insert(9, point) point = VirtualPoint("10.128.20.5", 10) avl.insert(10, point) point = VirtualPoint("10.128.20.6", 15) avl.insert(15, point) bal = avl.is_balanced() self.assertEqual(bal, True)
def test_avl_is_removing_and_balancing_correctly(self): """ Test that it is balancing correctly after removal of a node """ avl = AVLTree() point = VirtualPoint("10.128.20.1", 20) avl.insert(20, point) point = VirtualPoint("10.128.20.2", 4) avl.insert(4, point) point = VirtualPoint("10.128.20.3", 3) avl.insert(3, point) point = VirtualPoint("10.128.20.4", 9) avl.insert(9, point) point = VirtualPoint("10.128.20.5", 10) avl.insert(10, point) point = VirtualPoint("10.128.20.6", 15) avl.insert(15, point) avl.remove(20) # Will cause tree to rebalance bal = avl.is_balanced() self.assertEqual(bal, True)