Example #1
0
class TestAccount(unittest.TestCase):
    '''Test cases for the Account class'''

    def setUp(self):        
        self.acc = Account("me")
        
    def tearDown(self):
        pass
    
    def testCreate(self):
        assert self.acc._accountholder is not None

    def testDebitAndCredit(self):
        self.acc.credit(70)
        assert self.acc.getBalance() == 70
        self.acc.debit(70)
        assert self.acc.getBalance() == 0
        self.acc.credit(40)
        self.acc.credit(60)
        assert self.acc.getBalance() == 100

    def testExtractTooMuch(self):
        print 'Trying to extract too much'
        self.failUnlessRaises(InsufficientFunds, self.acc.debit, 70)