Fix tests

firedrakeproject · Dec 21, 2024 · 337d087 · 337d087
1 parent a86f3f5
commit 337d087
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/firedrake/assemble.py b/firedrake/assemble.py
@@ -817,8 +817,8 @@ def restructure_base_form(expr, visited=None):
             return ufl.action(expr, ustar)
 
         # -- Case (6) -- #
-        if isinstance(expr, ufl.FormSum) and all(not isinstance(c, ufl.form.BaseForm) for c in expr.components()):
-            # Return ufl.Sum
+        if isinstance(expr, ufl.FormSum) and all(ufl.duals.is_dual(a.function_space()) for a in expr.arguments()):
+            # Return ufl.Sum if we are assembling a FormSum with Coarguments (a primal expression)
             return sum(w*c for w, c in zip(expr.weights(), expr.components()))
         return expr