Código de Python - ¿VOCAL O NO?

# -*- coding: utf-8 -*-

"""

Created on Fri Mar 26 16:33:45 2021

@author: RubénColladoDíaz

"""

def vocal_o_no(x):

    if x == 'a' or x == 'e' or x == 'i' or x == 'u':

        return True

    else:

        return False

print(vocal_o_no('p'))