示例#1
0
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
示例#2
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
示例#3
0
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
示例#4
0
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
示例#5
0
def test_is_root_balanced():
    """Test root is balanced."""
    from bst import Tree
    test_tree = Tree([5])
    assert test_tree.balanced() == 0