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)