adding time tests example

Author: guduchango

src/bank_account.py

@@ -1,11 +1,12 @@
-from src.exceptions import InsufficientFundsError
+from datetime import datetime
+from src.exceptions import InsufficientFundsError, WithdrawalTimeRestrictionError
 
 
 class BankAccount:
     def __init__(self, balance=0, log_file=None):
         self.balance = balance
         self.log_file = log_file
-        self._log_transaction('Cuenta creada')
+        self._log_transaction("Cuenta creada")
 
     def _log_transaction(self, message):
         if self.log_file:
@@ -19,8 +20,14 @@ def deposit(self, amount):
         return self.balance
 
     def withdraw(self, amount):
+        now = datetime.now()
+        if now.hour < 8 or now.hour > 17:
+            raise WithdrawalTimeRestrictionError("Withdrawals are only allowed from 8am to 5pm")
+
         if amount > self.balance:
-            raise InsufficientFundsError(f"Withdrawal of {amount} exceeds balance {self.balance}")
+            raise InsufficientFundsError(
+                f"Withdrawal of {amount} exceeds balance {self.balance}"
+            )
         if amount > 0:
             self.balance -= amount
             self._log_transaction(f"Withdrew {amount}. New balance: {self.balance}")

src/exceptions.py

@@ -1,2 +1,5 @@
 class InsufficientFundsError(Exception):
+    pass
+
+class WithdrawalTimeRestrictionError(Exception):
     pass

tests/test_bank_account.py

@@ -57,18 +57,4 @@ def test_withdraw_disallow_before_bussines_hours(self, mock_datetime):
     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"])
+            self.account.withdraw(100)