"Enhanced Modular Arithmetic 'add' Test Cases

Expanded the test coverage for the 'add' method in the 'TestModularAdd' class. The changes introduce comprehensive test cases to cover scenarios including simple addition, modulus result, wrapping around modulus, edge cases, and handling of negative numbers. This level of thorough testing is aimed at ensuring correct and expected behavior of the 'add' method across an extensive range of inputs and situations."

Co-authored-by: Carlos González <[email protected]>

Author: albexl

tests/test_modular_arithmetics.py

@@ -5,17 +5,33 @@
 from algorithms.number_theory.modular_arithmetics import Modular
 
 
-class TestNaive(TestCase):
-    """Class to test the modular arithmetics implementation."""
+class TestModularAdd(TestCase):
+    """Class to test the 'add' method of modular arithmetic implementation."""
 
     def setUp(self):
         self.modular = Modular(10)
 
     def test_add(self):
-        """Test the `add` method."""
-        self.assertEqual(self.modular.add(3, 4), 7)
-        self.assertEqual(self.modular.add(5, 5), 0)
-        self.assertEqual(self.modular.add(8, 5), 3)
+        """Test the `add` method with various inputs."""
+        test_cases = [
+            (3, 4, 7),  # Simple addition
+            (5, 5, 0),  # Addition resulting in modulus
+            (8, 5, 3),  # Addition wrapping around modulus
+            (0, 0, 0),  # Edge case: adding zeros
+            (-1, 2, 1),  # Negative numbers
+            (10, 10, 0),  # Inputs equal to modulus
+        ]
+
+        for a, b, expected in test_cases:
+            with self.subTest(a=a, b=b, expected=expected):
+                self.assertEqual(self.modular.add(a, b), expected)
+
+
+class TestNaive(TestCase):
+    """Class to test the modular arithmetics implementation."""
+
+    def setUp(self):
+        self.modular = Modular(10)
 
     def test_sub(self):
         """Test the `sub` method."""