added test cases for ternary search

mans-andersson · Jan 28, 2021 · ef63e53 · ef63e53
1 parent 7a1e725
commit ef63e53
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/algorithms/search/__init__.py b/algorithms/search/__init__.py
@@ -1,4 +1,5 @@
 from .binary_search import *
+from .ternary_search import *
 from .first_occurrence import *
 from .last_occurrence import *
 from .linear_search import *

diff --git a/tests/test_search.py b/tests/test_search.py
@@ -1,5 +1,6 @@
 from algorithms.search import (
     binary_search, binary_search_recur,
+    ternary_search,
     first_occurrence,
     last_occurrence,
     linear_search,
@@ -41,6 +42,15 @@ def test_binary_search(self):
         self.assertEqual(11, binary_search_recur(array, 0, 11, 6))
         self.assertEqual(-1, binary_search_recur(array, 0, 11, 7))
         self.assertEqual(-1, binary_search_recur(array, 0, 11, -1))
+
+    def test_ternary_search(self):
+        array = [1, 2, 3, 3, 3, 3, 4, 4, 4, 4, 5, 6]
+        self.assertEqual(10, ternary_search(0, 11, 5, array))
+        self.assertEqual(3, ternary_search(0, 10, 3, array))
+        self.assertEqual(-1, ternary_search(0, 10, 5, array))
+        self.assertEqual(-1, ternary_search(0, 11, 7, array))
+        self.assertEqual(-1, ternary_search(0, 11, -1, array))
+
 
     def test_last_occurrence(self):
         array = [1, 2, 3, 3, 3, 3, 4, 4, 4, 4, 5, 6, 6, 6]