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"