class BankTestCase(unittest.TestCase): def setUp(self) -> None: self.bank = Bank('accounts.csv', 'transactions.csv') def test_get_account(self) -> None: self.assertIsInstance(self.bank.get_account(345), Account) def test_get_transactions(self) -> None: self.assertEqual(len(self.bank.get_transactions(345)), 2) def test_register_accounts(self) -> None: self.assertEqual(len(self.bank.accounts.items()), 1) def test_register_transactions(self) -> None: self.assertEqual(len(self.bank.transactions.items()), 1) def test_calculate_transactions(self) -> None: self.bank.calculate_balances() self.assertEqual(self.bank.get_account(345).balance, Decimal(-2500))
def main(args): bank = Bank(*args.files) bank.calculate_balances() bank.get_balances()