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"