def test_get_account_balance(self): bank = Bank() account_1 = Account(001, 50) bank.add_account(account_1) self.assertEqual(bank.get_account_balance(001), 50) self.assertEqual(bank.get_account_balance(002), 'Account does not exist')
def test_get_account_balance(self): bank = Bank() account1 = Account('001', 50) bank.add_account(account1) self.assertEqual(bank.get_account_balance('001'), 50)
def test_withdraw(self): bank = Bank() account_1 = Account(001, 50) bank.get_account_balance(account_1) self.assertEqual(bank.withdraw(001, 50, 40), 10)