Python Bank.calculate_balancesの例

コード例 #1

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))

コード例 #2

ファイル: rispatest.py プロジェクト: hugobrilhante/rispar

def main(args):
    bank = Bank(*args.files)
    bank.calculate_balances()
    bank.get_balances()