test: add parameterized tests

Author: BrianLusina

poetry.lock

@@ -8,6 +8,7 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "^3.11"
+parameterized = "^0.9.0"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.3.5"

pyproject.toml

@@ -35,7 +35,7 @@ def group_anagrams(strs: List[str]) -> List[List[str]]:
 
     Let’s see how we can implement the above algorithm:
 
-    - For each string, compute a 6-element list. Each element in this list represents the frequency of an English letter
+    - For each string, compute a 26-element list. Each element in this list represents the frequency of an English letter
      in the corresponding string. This frequency count will be represented as a tuple. For example, "abbccc" will be
      represented as (1, 2, 3, 0, 0, ..., 0). This mapping will generate identical lists for strings that are anagrams.
 
@@ -59,7 +59,7 @@ def group_anagrams(strs: List[str]) -> List[List[str]]:
     # traversing the list of strings takes O(n) time where n is the number of strings in this list
     for word in strs:
         count = [0] * 26
-        for char in word:
+        for char in word.lower():
             index = ord(char) - ord('a')
             count[index] += 1
 

pystrings/anagram/group_anagrams/__init__.py

@@ -1,8 +1,33 @@
 import unittest
-from . import group_anagrams
+from parameterized import parameterized
+from . import group_anagrams, group_anagrams_naive
 
 
 class GroupAnagramsTestCase(unittest.TestCase):
+    @parameterized.expand([
+        ("test_1", ["eat", "beat", "neat", "tea"], [["eat", "tea"], ["beat"], ["neat"]]),
+        ("test_2", ["duel", "dule", "speed", "spede", "deul", "cars"], [["duel", "dule", "deul"], ["speed", "spede"], ["cars"]]),
+        ("test_3", ["eat","tea","tan","ate","nat","bat"], [["eat","tea","ate"],["tan","nat"],["bat"]]),
+        ("test_4", ["word","sword","drow","rowd","iced","dice"], [["word","drow","rowd"],["sword"],["iced","dice"]]),
+        ("test_5", ["eat","drink","sleep","repeat"], [["eat"],["drink"],["sleep"],["repeat"]]),
+        ("test_6", ["hello","ohlle","dark"], [["hello","ohlle"],["dark"]])
+    ])
+    def test1(self, _, strs, expected):
+        actual = group_anagrams(strs)
+        self.assertEqual(actual, expected)
+
+    @parameterized.expand([
+        ("test_1", ["eat", "beat", "neat", "tea"], [["eat", "tea"], ["beat"], ["neat"]]),
+        ("test_2", ["duel", "dule", "speed", "spede", "deul", "cars"], [["duel", "dule", "deul"], ["speed", "spede"], ["cars"]]),
+        ("test_3", ["eat","tea","tan","ate","nat","bat"], [["eat","tea","ate"],["tan","nat"],["bat"]]),
+        ("test_4", ["word","sword","drow","rowd","iced","dice"], [["word","drow","rowd"],["sword"],["iced","dice"]]),
+        ("test_5", ["eat","drink","sleep","repeat"], [["eat"],["drink"],["sleep"],["repeat"]]),
+        ("test_6", ["hello","ohlle","dark"], [["hello","ohlle"],["dark"]])
+    ])
+    def test2(self, _, strs, expected):
+        actual = group_anagrams_naive(strs)
+        self.assertEqual(actual, expected)
+
     def test_1(self):
         strs = ["eat", "beat", "neat", "tea"]
         expected = [["eat", "tea"], ["beat"], ["neat"]]
@@ -39,12 +64,6 @@ def test_6(self):
         actual = group_anagrams(strs)
         self.assertEqual(expected, actual)
 
-    def test_7(self):
-        strs = ["eat","beat","neat","tea"]
-        expected = [["eat","tea"],["beat"],["neat"]]
-        actual = group_anagrams(strs)
-        self.assertEqual(expected, actual)
-
 
 if __name__ == '__main__':
     unittest.main()