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"