Function test_code_gen

tests/pytest/test_shapeoptimization.py:210–238

()

@pytest.mark.slow
def test_code_gen():
    mesh = Mesh(unit_cube.GenerateMesh(maxh=0.3))

    VEC = VectorH1(mesh,order=1)
    Phi,Psi = VEC.TnT()
    gfvh = GridFunction(VEC)
    gfvh.Set(CF( (x,y*x,x*y*z) ),definedon=mesh.Boundaries(".*"))
    gfh = GridFunction(H1(mesh,order=1))
    gfh.Set(x*y*z,definedon=mesh.Boundaries(".*"))

    gfhc = GridFunction(HCurl(mesh,order=1))
    gfhc.Set(CF( (x,y*x,x*y*z) ),definedon=mesh.Boundaries(".*"))
    gfhd = GridFunction(HDivSurface(mesh,order=1))
    gfhd.Set(CF( (x,y*x,x*y*z) ),definedon=mesh.Boundaries(".*"))
    

    n = specialcf.normal(3)

    functions = [(CF(1)*ds).DiffShape(Psi),(Trace(Grad(n))*ds).DiffShape(Psi),(CF(1)*ds(element_boundary=True)).DiffShape(Psi), (Trace(Grad(gfvh).Trace())*ds).DiffShape(Psi), ((Grad(gfh).Trace())**2*ds).DiffShape(Psi), (curl(gfhc).Trace()*gfhc.Trace()*ds).DiffShape(Psi),(div(gfhd).Trace()*gfhd.Trace()*gfhd.Trace()*ds).DiffShape(Psi)]


    for cf in functions:
        cfs = [ cf.Compile(), cf.Compile(True, wait=True)]
        for f in cfs:
            lf = BilinearForm(VEC)
            lf += Variation(cf-f)
            assert lf.Energy(gfvh.vec) == approx(0)
        
    return

if __name__ == "__main__":
    test_diff()