add xx,yy,zz ordering

mscroggs · Sep 15, 2023 · b57efd0 · b57efd0
1 parent fac4d3a
commit b57efd0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/symfem/finite_element.py b/symfem/finite_element.py
@@ -206,6 +206,11 @@ def tabulate_basis(
             for row in tabbed:
                 output.append(tuple(j for i in zip(*row) for j in i))
             return output
+        elif order == "xx,yy,zz":
+            output = []
+            for row in tabbed:
+                output.append(tuple(tuple(j for j in i) for i in zip(*row)))
+            return output
         elif order == "xyzxyz":
             output = []
             for row in tabbed: