예제 #1
0
def test_mirror():
    tree = Tree()
    l = [38, 13, 51, 10, 25, 40, 84, 12, 37, 66, 89, 95]
    length = len(l)
    while len(l) > 0:
        key = l.pop(0)
        value = chr(key)
        tree.insert(key, value)
    level_order = tree.level_order()
    assert level_order == [38, 13, 51, 10, 25, 40, 84, 12, 37, 66, 89, 95]
    tree.mirror()
    level_order = tree.level_order()
    pre_order = tree.pre_order()
    assert level_order == [38, 51, 13, 84, 40, 25, 10, 89, 66, 37, 12, 95]
    assert pre_order == [38, 51, 84, 89, 95, 66, 40, 13, 25, 37, 10, 12]
예제 #2
0
def test_in_order_pre_order_level_order():
    tree = Tree()
    l = [38, 13, 51, 10, 25, 40, 84, 12, 37, 66, 89, 95]
    length = len(l)
    while len(l) > 0:
        key = l.pop(0)
        value = chr(key)
        tree.insert(key, value)
    in_order = tree.in_order()
    pre_order = tree.pre_order()
    level_order = tree.level_order()
    assert in_order == [10, 12, 13, 25, 37, 38, 40, 51, 66, 84, 89, 95]
    assert pre_order == [38, 13, 10, 12, 25, 37, 51, 40, 84, 66, 89, 95]
    assert level_order == [38, 13, 51, 10, 25, 40, 84, 12, 37, 66, 89, 95]