def test_get_account_balance(self): bank = Bank() account_1 = Account(001, 75) bank.add_account(account_1) self.assertEqual(bank.get_account_balance(001), 75)
def test_add_account(self): bank = Bank() account_1 = Account(001, 50) account_2 = Account(002, 100) bank.add_account(account_1) bank.add_account(account_2) self.assertEqual(len(bank.accounts), 2)