def test_larger_balanced_tree(): """Test tree is balanced.""" from bst import Tree test_tree = Tree([10, 5, 15, 4, 6, 11, 16]) assert test_tree.balanced() == 0
def test_left_heavy_balanced_left(): """Test tree is left heavy.""" from bst import Tree test_tree = Tree([5, 4, 3, 2]) assert test_tree.balanced() == 1
def test_two_nodes_balanced(): """Test One root, and one left node.""" from bst import Tree test_tree = Tree([5, 1]) assert test_tree.balanced() == 1
def test_right_heavy_balanced_right(): """Test tree is right heavy.""" from bst import Tree test_tree = Tree([5, 1, 6, 7, 8]) assert test_tree.balanced() == -1
def test_is_root_balanced(): """Test root is balanced.""" from bst import Tree test_tree = Tree([5]) assert test_tree.balanced() == 0