def test_transaction_incorrect_choice():
    invalid_choice = ["aa", None]

    transaction = Transaction()

    for choice in invalid_choice:
        assert transaction.transaction_type(
            choice) == False, "Incorrect Choice"
def test_transaction_choice():
    valid_transaction_types = ["Withdrawal", "Deposit Cheque"]
    transaction = Transaction()

    for choice in valid_transaction_types:
        assert transaction.transaction_type(
            choice
        ) == True, "Please choose from the following two Transactions"