Skip to content

Commit b413fcc

Browse files
committed
Updated tests for searching
1 parent a286969 commit b413fcc

File tree

1 file changed

+21
-1
lines changed

1 file changed

+21
-1
lines changed

tests/test_searching.py

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@
44
linear_search,
55
binary_search,
66
breadth_first_search,
7-
depth_first_search)
7+
depth_first_search,
8+
exponential_search,
9+
interpolation_search,
10+
ternary_search)
811

912

1013
class TestSearchingAlgorithm(unittest.TestCase):
@@ -74,6 +77,23 @@ def test_dfs(self):
7477
result = depth_first_search.search(self.graph, 'G')
7578
self.assertEqual(result, ['G', 'C', 'A', 'B', 'D', 'F', 'E'])
7679

80+
class TestExponentialSearch(TestSearchingAlgorithm):
81+
def test_exponential_search(self):
82+
self.assertEqual(exponential_search.search(self.array, 7), 7)
83+
84+
alpha_result = linear_search.search(self.alphaArray, 'n')
85+
self.assertIs(alpha_result, 5)
86+
87+
class TestInterpolationSearch(TestSearchingAlgorithm):
88+
def test_interpolation_search(self):
89+
self.assertEqual(interpolation_search.search(self.array, 7), 7)
90+
91+
class TestTernarySearch(TestSearchingAlgorithm):
92+
def test_ternary_search(self):
93+
self.assertEqual(ternary_search.search(self.array, 0, len(self.array), 7), 7)
94+
95+
alpha_result = ternary_search.search(self.alphaArray, 0, len(self.alphaArray), 'n')
96+
self.assertIs(alpha_result, 5)
7797

7898
if __name__ == '__main__':
7999
unittest.main()

0 commit comments

Comments
 (0)