pytest example

Author: guduchango

requirements.txt

@@ -1,4 +1,5 @@
 ipdb==0.13.13
 requests==2.32.3
 Faker==33.3.0
-coverage==7.6.10
+coverage==7.6.10
+pytest==8.3.4

src/bank_account.py

@@ -14,6 +14,8 @@ def _log_transaction(self, message):
                 f.write(f"{message}\n")
 
     def deposit(self, amount):
+        if amount < 0:
+            raise ValueError("Amount cannot be negative")
         if amount > 0:
             self.balance += amount
             self._log_transaction(f"Deposited {amount}. New balance: {self.balance}")

src_tests/test_pytest.py

@@ -0,0 +1,20 @@
+import pytest
+from src.bank_account import BankAccount
+
+
+@pytest.mark.parametrize("ammount, expected", [
+    (100, 1100),
+    (3000, 4000),
+    (4500, 5500),
+])
+def test_deposit_multiple_ammounts(ammount, expected):
+    account = BankAccount(balance=1000, log_file="transactions.txt")
+    new_balance = account.deposit(ammount)
+    assert new_balance == expected
+
+
+# TODO: Create a test that does't allow the deposit to be less than zero
+def test_deposit_less_than_zero():
+    account = BankAccount(balance=1000, log_file="transactions.txt")
+    with pytest.raises(ValueError):
+        account.deposit(-100)