Merge branch 'master' into pbrubeck/hhj

firedrakeproject · Nov 18, 2024 · 8d69f5c · 8d69f5c
2 parents 359fbc4 + e06308d
commit 8d69f5c
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 9 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -15,7 +15,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python_version: ['3.8', '3.9', '3.10', '3.11']
+        python_version: ['3.10', '3.11', '3.12', '3.13']
       # Don't immediately kill all if one Python version fails
       fail-fast: false
     steps:

diff --git a/tests/test_interpolation_factorisation.py b/tests/test_interpolation_factorisation.py
@@ -54,7 +54,7 @@ def test_sum_factorisation_scalar_tensor(mesh, element):
     source = element(degree - 1)
     target = element(degree)
     tensor_flops = flop_count(mesh, source, target)
-    expect = numpy.prod(target.value_shape)
+    expect = FunctionSpace(mesh, target).value_size
     if isinstance(target, FiniteElement):
         scalar_flops = tensor_flops
     else:

diff --git a/tests/test_tsfc_204.py b/tests/test_tsfc_204.py
@@ -13,8 +13,8 @@ def test_physically_mapped_facet():
     V = FiniteElement("P", mesh.ufl_cell(), 1)
     R = FiniteElement("P", mesh.ufl_cell(), 1)
     Vv = VectorElement(BrokenElement(V))
-    Qhat = VectorElement(BrokenElement(V[facet]))
-    Vhat = VectorElement(V[facet])
+    Qhat = VectorElement(BrokenElement(V[facet]), dim=2)
+    Vhat = VectorElement(V[facet], dim=2)
     Z = FunctionSpace(mesh, MixedElement(U, Vv, Qhat, Vhat, R))
 
     z = Coefficient(Z)

diff --git a/tsfc/kernel_interface/common.py b/tsfc/kernel_interface/common.py
@@ -472,7 +472,7 @@ def prepare_coefficient(coefficient, name, interior_facet=False):
 
     if coefficient.ufl_element().family() == 'Real':
         # Constant
-        value_size = coefficient.ufl_element().value_size
+        value_size = coefficient.ufl_function_space().value_size
         expression = gem.reshape(gem.Variable(name, (value_size,)),
                                  coefficient.ufl_shape)
         return expression

diff --git a/tsfc/ufl_utils.py b/tsfc/ufl_utils.py
@@ -416,8 +416,9 @@ def apply_mapping(expression, element, domain):
         mesh = domain
     if domain is not None and mesh != domain:
         raise NotImplementedError("Multiple domains not supported")
-    if expression.ufl_shape != element.value_shape:
-        raise ValueError(f"Mismatching shapes, got {expression.ufl_shape}, expected {element.value_shape}")
+    pvs = element.pullback.physical_value_shape(element, mesh)
+    if expression.ufl_shape != pvs:
+        raise ValueError(f"Mismatching shapes, got {expression.ufl_shape}, expected {pvs}")
     mapping = element.mapping().lower()
     if mapping == "identity":
         rexpression = expression
@@ -462,7 +463,7 @@ def apply_mapping(expression, element, domain):
         sub_elem = element.sub_elements[0]
         shape = expression.ufl_shape
         flat = ufl.as_vector([expression[i] for i in numpy.ndindex(shape)])
-        vs = sub_elem.value_shape
+        vs = sub_elem.pullback.physical_value_shape(sub_elem, mesh)
         rvs = sub_elem.reference_value_shape
         seen = set()
         rpieces = []
@@ -483,7 +484,7 @@ def apply_mapping(expression, element, domain):
         # And reshape
         rexpression = as_tensor(numpy.asarray(rpieces).reshape(element.reference_value_shape))
     else:
-        raise NotImplementedError(f"Don't know how to handle mapping type {mapping} for expression of rank {element.value_shape}")
+        raise NotImplementedError(f"Don't know how to handle mapping type {mapping} for expression of rank {ufl.FunctionSpace(mesh, element).value_shape}")
     if rexpression.ufl_shape != element.reference_value_shape:
         raise ValueError(f"Mismatching reference shapes, got {rexpression.ufl_shape} expected {element.reference_value_shape}")
     return rexpression