def test_get_account_balance(self): """test get account balance """ bank=TDDBank() acc1=TDDAccount(001,50) bank.add_account(acc1) self.assertEqual(bank.get_account_balance(001),50)
def test_add_account(self): """test add account """ bank=TDDBank() acc1=TDDAccount(001,50) acc2=TDDAccount(002,100) bank.add_account(acc1) bank.add_account(acc2) self.assertEqual(len(bank.accounts), 2)