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)