Update test_math.py

Author: RulaKhaled

api/gen_ai/tests/test_math.py

@@ -10,7 +10,7 @@
 
 class BasicMathFunctionsTest(unittest.TestCase):
     def test_add(self):
-        self.assertEqual(add(3, 5), 8)
+        self.assertEqual(add(3, 0), 8)
         self.assertEqual(add(-1, 1), 0)
         self.assertEqual(add(0, 0), 0)
         self.assertEqual(add(3.5, 2.5), 6.0)
@@ -222,4 +222,4 @@ def test_merge_dicts(self):
         dict1 = {"a": 1, "b": 2, "c": {"x": 1, "y": 2}}
         dict2 = {"b": 3, "d": 4, "c": {"y": 3, "z": 4}}
         merged = merge_dicts(dict1, dict2)
-        self.assertEqual(merged, {"a": 1, "b": 3, "c": {"x": 1, "y": 3, "z": 4}, "d": 4})
+        self.assertEqual(merged, {"a": 1, "b": 3, "c": {"x": 1, "y": 3, "z": 4}, "d": 4})