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)