Cómo pasar la función como paraemter de otra función pythpn

def add(a, b):
    return a + b

def mul(a, b):
    return a * b

def calculate(a, b, func): #calculate is higher order function
    return func(a, b)

print(calculate(2, 3, mul))
Arpan