Reimplement TensorElement support
We previously supported ufl.TensorElement
, but this got lost in the new-style handling of systems. The crux is that the additional shape of the tensorelement needs to be mapped to a flat VectorElement at code generation time. The relevant (currently disabled) tests are stokes_stress_*
.