Update firedrake/output/vtk_output.py

Co-authored-by: Connor Ward <[email protected]>
firedrakeproject · Nov 29, 2024 · 21715bd · 21715bd
1 parent b3d8705
commit 21715bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/firedrake/output/vtk_output.py b/firedrake/output/vtk_output.py
@@ -479,7 +479,8 @@ def _prepare_output(self, function, max_elem):
             raise ValueError("Unsupported shape %s" % (shape, ))
         if isinstance(function, Function):
             output = Function(V)
-        elif isinstance(function, Cofunction):
+        else:
+            assert isinstance(function, Cofunction)
             output = Function(V.dual())
 
         if self.project: