def test_print_three_nodes_even(): x = AVLNode() x.value = 10 x.left_node = AVLNode() x.left_node.value = 5 x.right_node = AVLNode() x.right_node.value = 15 print x.to_string() assert x.to_string( ) == "Value: 10\nLeft node->Value: 5\nRight node->Value: 15"
def test_print_two_nodes_right(): x = AVLNode() x.value = 10 x.right_node = AVLNode() x.right_node.value = 15 print x.to_string() assert x.to_string() == "Value: 10\nRight node->Value: 15"
def test_print_two_nodes_left(): x = AVLNode() x.value = 10 x.left_node = AVLNode() x.left_node.value = 5 print x.to_string() assert x.to_string() == "Value: 10\nLeft node->Value: 5"
def test_node_rebalance_right_left_balance(): x = AVLTree() x.head_node = AVLNode() x.add(10) x.add(15) x.add(13) print x.to_string() assert x.head_node.balance is 0
def test_node_rebalance_right_right_structure(): x = AVLTree() x.head_node = AVLNode() x.add(1) x.add(5) x.add(10) assert x.head_node.value is 5 assert x.head_node.left_node.value is 1 assert x.head_node.right_node.value is 10
def test_node_rebalance_right_left_structure(): x = AVLTree() x.head_node = AVLNode() x.add(10) x.add(15) x.add(13) print x.to_string() assert x.head_node.value is 13 assert x.head_node.left_node.value is 10 assert x.head_node.right_node.value is 15
def test_print_three_nodes_even(): x = AVLNode() x.value = 10 x.left_node = AVLNode() x.left_node.value = 5 x.right_node = AVLNode() x.right_node.value = 15 print x.to_string() assert x.to_string() == "Value: 10\nLeft node->Value: 5\nRight node->Value: 15"
def test_node_rebalance_right_right_depth(): x = AVLTree() x.head_node = AVLNode() x.add(1) x.add(5) x.add(10) assert x.head_node.right_depth is 1 assert x.head_node.left_depth is 1 assert x.head_node.right_node.left_depth is 0 assert x.head_node.right_node.right_depth is 0 assert x.head_node.left_node.left_depth is 0 assert x.head_node.left_node.right_depth is 0
def test_print_four_nodes(): x = AVLNode() x.value = 10 x.left_node = AVLNode() x.left_node.value = 5 x.right_node = AVLNode() x.right_node.value = 15 x.left_node.left_node = AVLNode() x.left_node.left_node.value = 1 print x.to_string() assert x.to_string() == "Value: 10\nLeft node->Value: 5\nLeft node->Value: 1\nRight node->Value: 15"
def test_node_rebalance_right_left_depth(): x = AVLTree() x.head_node = AVLNode() x.add(10) x.add(15) x.add(13) print x.to_string() assert x.head_node.right_depth is 1 assert x.head_node.left_depth is 1 assert x.head_node.right_node.left_depth is 0 assert x.head_node.right_node.right_depth is 0 assert x.head_node.left_node.left_depth is 0 assert x.head_node.left_node.right_depth is 0
def test_print_single_node(): x = AVLNode() x.value = 10 assert x.to_string() == "Value: 10"
def test_print_four_nodes(): x = AVLNode() x.value = 10 x.left_node = AVLNode() x.left_node.value = 5 x.right_node = AVLNode() x.right_node.value = 15 x.left_node.left_node = AVLNode() x.left_node.left_node.value = 1 print x.to_string() assert x.to_string( ) == "Value: 10\nLeft node->Value: 5\nLeft node->Value: 1\nRight node->Value: 15"