Update integration domains on vform

src/bvpy/templates/vforms/elasticity.py

@@ -163,9 +163,9 @@ class LinearElasticForm(ElasticForm):
 
     def construct_form(self, u, v, sol):
         self.lhs = fe.Constant(self._thickness)\
-                   * fe.inner(self._stress(u), self._strain(v)) * fe.dx
+                   * fe.inner(self._stress(u), self._strain(v)) * self.dx
 
-        self.rhs = fe.dot(self._source_term, v) * fe.dx(domain=v.function_space().mesh())\
+        self.rhs = fe.dot(self._source_term, v) * self.dx(domain=v.function_space().mesh())\
                    + self._neumann_term(v
 
 
@@ -262,8 +262,8 @@ class HyperElasticForm(ElasticForm):
             I1 = fe.tr(C) * J**(-2 / 3)
             W = (self._mu / 2) * (I1 - 3) + (K / 4) * ((J - 1)**2 + (fe.ln(J))**2)
 
-        self.lhs = self._thickness * W * fe.dx
-        self.lhs -= fe.dot(self._source_term, sol) * fe.dx
+        self.lhs = self._thickness * W * self.dx
+        self.lhs -= fe.dot(self._source_term, sol) * self.dx
         self.lhs -= self._neumann_term(v)
 
         self.lhs = fe.derivative(self.lhs, sol, v)

src/bvpy/templates/vforms/poisson.py

@@ -39,8 +39,8 @@ class PoissonForm(AbstractVform):
             self._source_term = source
 
     def construct_form(self, u, v, sol):
-        self.lhs = fe.inner(fe.grad(u), fe.grad(v))*fe.dx
-        self.rhs = self._source_term*v*fe.dx + self._neumann_term(v)
+        self.lhs = fe.inner(fe.grad(u), fe.grad(v))*self.dx
+        self.rhs = self._source_term*v*self.dx + self._neumann_term(v)
         # self.rhs = self._source_term*v*fe.dx + self._neuman*v*self.ds(0)