ex18 as exx27, 30 #534

kinnala · Jan 7, 2021 · 0afa4ff · 0afa4ff
1 parent f976251
commit 0afa4ff
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/docs/examples/ex18.py b/docs/examples/ex18.py
@@ -78,11 +78,9 @@ def body_force(v, w):
 
 velocity, pressure = np.split(uvp, [A.shape[0]])
 
-basis['psi'] = InteriorBasis(mesh, ElementTriP2())
+basis['psi'] = InteriorBasis(mesh, ElementTriP2(), quadrature=basis['u'].quadrature)
 A = asm(laplace, basis['psi'])
-vorticity = asm(rot, basis['psi'],
-                w=[basis['psi'].interpolate(velocity[i::2])
-                   for i in range(2)])
+vorticity = asm(rot, basis['psi'], w=basis['u'].interpolate(velocity))
 psi = solve(*condense(A, vorticity, D=basis['psi'].find_dofs()))