Logic fix

FEniCS · Nov 19, 2024 · fbb37f8 · fbb37f8
1 parent a2f6235
commit fbb37f8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/python/basix/ufl.py b/python/basix/ufl.py
@@ -2237,7 +2237,7 @@ def mixed_element(elements: list[_ElementBase]) -> _ElementBase:
     if len(elements) > 1:
         return _MixedElement(elements)
     else:
-        return elements
+        return elements[0]
 
 
 def quadrature_element(

diff --git a/test/test_ufl_wrapper.py b/test/test_ufl_wrapper.py
@@ -98,11 +98,23 @@ def test_enriched_element(elements):
     "e,space0,space1",
     [
         (basix.ufl.element("Lagrange", basix.CellType.triangle, 2), "H1", basix.SobolevSpace.H1),
+        (
+            basix.ufl.mixed_element([basix.ufl.element("Lagrange", basix.CellType.triangle, 2)]),
+            "H1",
+            basix.SobolevSpace.H1,
+        ),
         (
             basix.ufl.element("Discontinuous Lagrange", basix.CellType.triangle, 0),
             "L2",
             basix.SobolevSpace.L2,
         ),
+        (
+            basix.ufl.mixed_elementelement(
+                [basix.ufl.element("Discontinuous Lagrange", basix.CellType.triangle, 0)]
+            ),
+            "L2",
+            basix.SobolevSpace.L2,
+        ),
         (
             basix.ufl.mixed_element(
                 [