Add element dtype

jorgensd · May 5, 2024 · b18f4a2 · b18f4a2
1 parent 504de00
commit b18f4a2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/tests/test_checkpointing.py b/tests/test_checkpointing.py
@@ -87,6 +87,7 @@ def test_read_write_P_3D(
         degree,
         basix.LagrangeVariant.gll_warped,
         shape=(mesh.geometry.dim,),
+        dtype=mesh.geometry.x.dtype,
     )
 
     def f(x):
@@ -178,6 +179,7 @@ def test_read_write_P_3D_time(
         degree,
         basix.LagrangeVariant.gll_warped,
         shape=(mesh.geometry.dim,),
+        dtype=mesh.geometry.x.dtype,
     )
 
     def f(x):

diff --git a/tests/test_checkpointing_vector.py b/tests/test_checkpointing_vector.py
@@ -56,7 +56,7 @@ def test_read_write_2D(
 ):
     mesh = simplex_mesh_2D
     f_dtype = get_dtype(mesh.geometry.x.dtype, is_complex)
-    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree)
+    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree, dtype=mesh.geometry.x.dtype)
 
     def f(x):
         values = np.empty((2, x.shape[1]), dtype=f_dtype)
@@ -81,7 +81,7 @@ def test_read_write_3D(
 ):
     mesh = simplex_mesh_3D
     f_dtype = get_dtype(mesh.geometry.x.dtype, is_complex)
-    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree)
+    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree, dtype=mesh.geometry.x.dtype)
 
     def f(x):
         values = np.empty((3, x.shape[1]), dtype=f_dtype)
@@ -114,7 +114,7 @@ def test_read_write_2D_quad(
 ):
     mesh = non_simplex_mesh_2D
     f_dtype = get_dtype(mesh.geometry.x.dtype, is_complex)
-    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree)
+    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree, dtype=mesh.geometry.x.dtype)
 
     def f(x):
         values = np.empty((2, x.shape[1]), dtype=f_dtype)
@@ -146,7 +146,7 @@ def test_read_write_hex(
 ):
     mesh = non_simplex_mesh_3D
     f_dtype = get_dtype(mesh.geometry.x.dtype, is_complex)
-    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree)
+    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree, dtype=mesh.geometry.x.dtype)
 
     def f(x):
         values = np.empty((3, x.shape[1]), dtype=f_dtype)
@@ -179,7 +179,7 @@ def test_read_write_multiple(
 ):
     mesh = non_simplex_mesh_2D
     f_dtype = get_dtype(mesh.geometry.x.dtype, is_complex)
-    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree)
+    el = basix.ufl.element(family, mesh.ufl_cell().cellname(), degree, dtype=mesh.geometry.x.dtype)
 
     def f(x):
         values = np.empty((2, x.shape[1]), dtype=f_dtype)