Clase 14: Uso de doctest

src/calculator.py

@@ -1,4 +1,11 @@
 def sum(a, b):
+    """
+    >>> sum(5, 7)
+    12
+
+    >>> sum(4, -4)
+    0
+    """
     return a + b
 
 
@@ -11,6 +18,12 @@ def multiply(a, b):
 
 
 def divide(a, b):
+    """
+    >>> divide(10, 0)
+    Traceback (most recent call last):
+    ValueError: La división por cero no está permitida
+    """
+
     if b == 0:
         raise ValueError("La división por cero no está permitida")
     return a / b

tests/test_bank_account.py

@@ -58,3 +58,17 @@ def test_withdraw_disallow_after_bussines_hours(self, mock_datetime):
         mock_datetime.now.return_value.hour = 18
         with self.assertRaises(WithdrawalTimeRestrictionError):
             self.account.withdraw(100)
+
+
+    def test_deposit_multiple_ammounts(self):
+
+        test_cases = [
+            {"ammount": 100, "expected": 1100},
+            {"ammount": 3000, "expected": 4000},
+            {"ammount": 4500, "expected": 5500},
+        ]
+        for case in test_cases:
+            with self.subTest(case=case):
+                self.account = BankAccount(balance=1000, log_file="transactions.txt")
+                new_balance = self.account.deposit(case["ammount"])
+                self.assertEqual(new_balance, case["expected"])