class TestAccount(unittest.TestCase): def setUp(self): self.transactions = [Transaction(345, -1000), Transaction(345, 1500)] self.account = Account(345, 14428) def test_init_error_number_value_empty(self): with self.assertRaises(ValueError): Account('', '14428') def test_init_error_balance_value_empty(self): with self.assertRaises(ValueError): Account('345', '') def test_calculate_balance(self): self.account.calculate_balance(self.transactions) self.assertEqual(self.account.balance, 14928) def test_calculate_balance_negative_balance(self): transactions = [Transaction(345, -1000), Transaction(345, -1600)] account = Account(345, 2500) account.calculate_balance(transactions) self.assertEqual(account.balance, -600) def test_create_accounts_list(self): rows = File.read_csv(f'{CSV_FOLDER}/accounts.csv') accounts = Account.create_accounts_list(rows) self.assertEqual(len(accounts), 2)
def test_calculate_balance_negative_balance(self): transactions = [Transaction(345, -1000), Transaction(345, -1600)] account = Account(345, 2500) account.calculate_balance(transactions) self.assertEqual(account.balance, -600)