Update test_all_permutations.py

Author: djeada

src/backtracking/python/all_permutations/tests/test_all_permutations.py

@@ -1,26 +1,51 @@
-from src.all_permutations import all_permutations
+# tests/test_all_permutations.py
+
 import unittest
 
+from src.all_permutations import (
+    all_permutations_itertools,
+    all_permutations_backtracking,
+    all_permutations  # this is aliased to backtracking by default
+)
+
 
 class TestAllPermutations(unittest.TestCase):
+    def setUp(self):
+        # We will test these two implementations side by side in every test.
+        self.funcs_to_test = [
+            all_permutations_itertools,
+            all_permutations_backtracking,
+        ]
+
     def test_two_elements(self):
         input_list = [1, 2]
-        excepted = [[1, 2], [2, 1]]
-        actual = all_permutations(input_list)
+        expected = [[1, 2], [2, 1]]
 
-        self.assertListEqual(sorted(excepted), sorted(actual))
+        for func in self.funcs_to_test:
+            with self.subTest(func=func.__name__):
+                actual = func(input_list[:])  # pass a fresh copy to avoid side‐effects
+                # Sort both lists of lists before comparing
+                self.assertListEqual(sorted(expected), sorted(actual))
 
     def test_three_elements(self):
         input_list = [3, 1, 2]
-        excepted = [[1, 2, 3], [1, 3, 2], [2, 1, 3], [2, 3, 1], [3, 2, 1], [3, 1, 2]]
-
-        actual = all_permutations(input_list)
+        expected = [
+            [1, 2, 3],
+            [1, 3, 2],
+            [2, 1, 3],
+            [2, 3, 1],
+            [3, 2, 1],
+            [3, 1, 2],
+        ]
 
-        self.assertListEqual(sorted(excepted), sorted(actual))
+        for func in self.funcs_to_test:
+            with self.subTest(func=func.__name__):
+                actual = func(input_list[:])
+                self.assertListEqual(sorted(expected), sorted(actual))
 
     def test_three_strings(self):
         input_list = ["A", "B", "C"]
-        excepted = [
+        expected = [
             ["A", "B", "C"],
             ["A", "C", "B"],
             ["B", "A", "C"],
@@ -29,13 +54,14 @@ def test_three_strings(self):
             ["C", "B", "A"],
         ]
 
-        actual = all_permutations(input_list)
-
-        self.assertListEqual(sorted(excepted), sorted(actual))
+        for func in self.funcs_to_test:
+            with self.subTest(func=func.__name__):
+                actual = func(input_list[:])
+                self.assertListEqual(sorted(expected), sorted(actual))
 
     def test_four_elements(self):
         input_list = [3, 1, 2, 4]
-        excepted = [
+        expected = [
             [3, 1, 2, 4],
             [3, 1, 4, 2],
             [3, 2, 1, 4],
@@ -62,9 +88,10 @@ def test_four_elements(self):
             [4, 2, 1, 3],
         ]
 
-        actual = all_permutations(input_list)
-
-        self.assertListEqual(sorted(excepted), sorted(actual))
+        for func in self.funcs_to_test:
+            with self.subTest(func=func.__name__):
+                actual = func(input_list[:])
+                self.assertListEqual(sorted(expected), sorted(actual))
 
 
 if __name__ == "__main__":