adding more tests, using class.

Author: krsm

Unitests/pystest_stdy/bank.py

@@ -0,0 +1,24 @@
+# -*- coding: utf-8 -*-
+"""
+
+Simple study related to pytest
+
+"""
+
+
+class InsufficientAmount(Exception):
+    pass
+
+
+class BankAccount(object):
+
+    def __init__(self, initial_amount=0):
+        self.current_balance = initial_amount
+
+    def withdrawal_cash(self, amount):
+        if self.current_balance < amount:
+            raise InsufficientAmount("Not enough cash to spend!{}".format(amount))
+        self.current_balance -= amount
+
+    def deposit_cash(self, amount):
+        self.current_balance += amount

Unitests/pystest_stdy/test_bank.py

@@ -0,0 +1,37 @@
+# -*- coding: utf-8 -*-
+"""
+
+Simple study related to pytest
+
+"""
+
+import pytest
+from Unitests.pystest_stdy.bank import BankAccount, InsufficientAmount
+
+
+def test_default_initial_amount():
+    bank_account = BankAccount()
+    assert bank_account.current_balance == 0
+
+
+def test_setting_initial_amount():
+    bank_account = BankAccount(100)
+    assert bank_account.current_balance == 100
+
+
+def test_bank_account_deposit_cash():
+    bank_account = BankAccount(100)
+    bank_account.deposit_cash(90)
+    assert bank_account.current_balance == 190
+
+
+def test_bank_account_withdrawal_cash():
+    bank_account = BankAccount(100)
+    bank_account.withdrawal_cash(90)
+    assert bank_account.current_balance == 10
+
+
+def test_bank_account_raises_exception():
+    bank_account = BankAccount()
+    with pytest.raises(InsufficientAmount):
+        bank_account.withdrawal_cash(100)