ya ingresado el producto a vender, el costo en listas, permitame ingresar la canitdad y que me evalue el costo segun la cantidad del producto

def ventas(costo, cantidad):
    if cantidad >= 100:
        return costo * cantidad * 0.95
    elif cantidad > 50:
        return costo * cantidad * 0.98
    return costo * cantidad