def test_search_mulitple_nodes():
"""Test multiple node searches."""
from bst import Tree
test_tree = Tree()
test_values = random.sample(range(100), 10)
for i in test_values:
test_tree.insert(i)
for j in test_values:
assert test_tree.search(j).data == j
def test_search_one_node():
"""Test search for one node from inserted list of nodes."""
from bst import Tree
test_tree = Tree()
test_node_list = [1, 2, 3, 4, 5]
for i in test_node_list:
test_tree.insert(i)
assert test_tree.search(test_node_list[2]).data == 3
