def test_find_rand_value(self):
bst = Tree()
for i in [[10, 1], [12, 2], [11, 3], [14, 4], [13, 5]]:
bst.insert(i)
for j in range(100):
search_for = random.randint(10, 14)
r = bst.search(search_for)
if search_for == 10:
s = [10, 1]
elif search_for == 11:
s = [11, 3]
elif search_for == 12:
s = [12, 2]
elif search_for == 13:
s = [13, 5]
elif search_for == 14:
s = [14, 4]
self.assertEqual(r, s)
def test_find_same_value(self):
bst = Tree()
for i in [[10, 1], [12, 2], [12, 3], [17, 3], [14, 4], [15, 5]]:
bst.insert(i)
r = bst.search(12)
self.assertEqual(r, [12, 2, 3])
def test_cannot_find_value(self):
bst = Tree()
for i in [[10, 1], [12, 2], [17, 3], [14, 4], [15, 5]]:
bst.insert(i)
r = bst.search(20)
self.assertEqual(r, False)
