Merge pull request #9 from CatalinAnt/Almu

Almu

Author: MuthanaMustafa

tests/test_dp.py

@@ -14,7 +14,8 @@
     longest_increasing_subsequence_optimized,
     longest_increasing_subsequence_optimized2,
     int_divide,find_k_factor,
-    planting_trees, regex_matching
+    planting_trees, regex_matching,
+    word_break,
 )
 
 
@@ -258,6 +259,16 @@ def test_symbol_2(self):
         p = "ab*"
         self.assertTrue(regex_matching.is_match(s, p))
 
+class TestWordBreak(unittest.TestCase):
+    """Test for interpolation_search and word_break"""
+
+    def test_word_break(self):
+        self.assertTrue(word_break("keonkim", {"keon", "kim"}))
+        self.assertTrue(word_break("leetcode", {"leet", "code"}))
+        self.assertFalse(word_break("catsandog", {"cats", "dog", "sand", "and", "cat"}))
+        self.assertTrue(word_break("applepenapple", {"apple", "pen"}))
+        self.assertFalse(word_break("pineapplepenapple", {"apple", "pen"}))
+        self.assertTrue(word_break("aaaaaaa", {"aaaa", "aaa"}))
 
 if __name__ == '__main__':
     unittest.main()

tests/test_sort.py

@@ -17,7 +17,8 @@
     radix_sort,
     gnome_sort,
     cocktail_shaker_sort,
-    top_sort, top_sort_recursive
+    top_sort, top_sort_recursive,
+    stoogesort
 )
 
 import unittest
@@ -124,7 +125,27 @@ def test_topsort(self):
         self.assertTrue(res.index('g') < res.index('e'))
         res = top_sort(self.depGraph)
         self.assertTrue(res.index('g') < res.index('e'))
-
+class TestStoog(unittest.TestCase):
+    def test_stoogesort(self):
+        arr1 = [1, 3, 64, 5, 7, 8]
+        stoogesort(arr1, 0, len(arr1) - 1)
+        self.assertEqual(arr1, [1, 3, 5, 7, 8, 64])
+
+        arr2 = [5, 4, 3, 2, 1]
+        stoogesort(arr2, 0, len(arr2) - 1)
+        self.assertEqual(arr2, [1, 2, 3, 4, 5])
+
+        arr3 = [1, 2, 3, 4, 5]
+        stoogesort(arr3, 0, len(arr3) - 1)
+        self.assertEqual(arr3, [1, 2, 3, 4, 5])
+
+        arr4 = [10, -1, 2, 3, 0]
+        stoogesort(arr4, 0, len(arr4) - 1)
+        self.assertEqual(arr4, [-1, 0, 2, 3, 10])
+
+        arr5 = []
+        stoogesort(arr5, 0, len(arr5) - 1)
+        self.assertEqual(arr5, [])
 
 if __name__ == "__main__":
     unittest.main()