Постобработка FEniCS: интегрировать по граничному произведению функций

Question

Из главы 5 "Решение PDE в Python" я понял, что могу вычислять поток

n = FacetNormal(mesh)
flux = -k*dot(grad(u), n)*ds
total_flux = assemble(flux)

В вышеупомянутом примере k постоянная Как я могу распространить это на случай, когда k функция как таковая

def k(x):
    # big complicated computation
    return result