def test_account_balance_after_transaction_is_empty():
balance_after_transaction_empty = ["0", None]
transaction = Transaction()
for balance in balance_after_transaction_empty:
assert transaction.account_balance_post_transaction(
balance) == False, "Account balance is now empty"
def test_account_balance_after_transaction():
balance_after_transaction = [
"1",
"10000",
"25000",
"45982",
"4",
]
transaction = Transaction()
for balance in balance_after_transaction:
assert transaction.account_balance_post_transaction(
balance) == True, "Account Balance is displayed"