잘 작동하는지 살펴보고자 테스트 코드를 작성해 보겠습니다. 먼저 트리를 하나 만들도록 하죠. 그림 7-5의 트리를 만들고 전위 순회를 통해 순회하도록 하겠습니다. 실행 결과로 1-2-4-5-3-6-7이 나오는지 확인해 봅시다.
코드 7-3
if __name__ == "__main__":
n1 = TreeNode(1)
n2 = TreeNode(2)
n3 = TreeNode(3)
n4 = TreeNode(4)
n5 = TreeNode(5)
n6 = TreeNode(6)
n7 = TreeNode(7)
n1.left = n2; n1.right = n3
n2.left = n4; n2.right = n5
n3.left = n6; n3.right = n7
preorder(n1)
print()
코드 7-3에서 노드를 일곱 개 만들어 그림 7-5의 노드를 만들었습니다. 그 후 루트 노드인 n1을 전달해서 preorder를 호출합니다.
1 2 4 5 3 6 7