def test_in_order_one(): """Test sort one node.""" from bst import Tree test_list = [10] test_tree = Tree(test_list) path = test_tree.in_order(test_tree.root) count = 0 for i in path: assert i.data == test_list[count] count += 1
def test_in_order_multiple_nodes(): """Test sort multiple nodes.""" from bst import Tree test_list = [5, 2, 6, 1, 4] test_tree = Tree(test_list) path = test_tree.in_order(test_tree.root) test_results = [1, 2, 4, 5, 6] count = 0 for i in path: assert i.data == test_results[count] count += 1