Merge pull request #72 from minrk/xdmf-tests-complex

fixes for xdmf complex tests
scientificcomputing · Nov 20, 2024 · 706c719 · 706c719
2 parents 34d6e0f + ddc47c5
commit 706c719
Showing 1 changed file with 19 additions and 14 deletions.
diff --git a/tests/test_xdmf.py b/tests/test_xdmf.py
@@ -6,6 +6,7 @@
 import numpy as np
 import scifem
 import pytest
+from petsc4py import PETSc
 from unittest.mock import patch
 
 try:
@@ -16,6 +17,10 @@
     hash5py = False
 
 
+stype = PETSc.ScalarType
+rtype = PETSc.RealType
+
+
 @pytest.mark.parametrize("use_ctx_manager", [True, False])
 @pytest.mark.parametrize(
     "backend",
@@ -38,9 +43,9 @@ def test_XDMFFile_2D(cell_type, degree, value_shape, backend, use_ctx_manager, t
         scheme="default", degree=degree, cell=mesh.ufl_cell().cellname(), value_shape=value_shape
     )
     V = dolfinx.fem.functionspace(mesh, el)
-    u = dolfinx.fem.Function(V)
-    v = dolfinx.fem.Function(V)
-    t = dolfinx.fem.Constant(mesh, 0.0)
+    u = dolfinx.fem.Function(V, dtype=stype)
+    v = dolfinx.fem.Function(V, dtype=stype)
+    t = dolfinx.fem.Constant(mesh, stype(0.0))
 
     X = ufl.SpatialCoordinate(mesh)
     if value_shape == ():
@@ -58,22 +63,22 @@ def test_XDMFFile_2D(cell_type, degree, value_shape, backend, use_ctx_manager, t
         with scifem.xdmf.XDMFFile(folder / "data.xdmf", [u, v], backend=backend) as xdmf:
             u.interpolate(dolfinx.fem.Expression(u_expr, V.element.interpolation_points()))
             v.interpolate(dolfinx.fem.Expression(v_expr, V.element.interpolation_points()))
-            xdmf.write(t.value)
+            xdmf.write(rtype(t.value))
 
             t.value = 0.3
             u.interpolate(dolfinx.fem.Expression(u_expr, V.element.interpolation_points()))
             v.interpolate(dolfinx.fem.Expression(v_expr, V.element.interpolation_points()))
-            xdmf.write(t.value)
+            xdmf.write(rtype(t.value))
     else:
         xdmf = scifem.xdmf.XDMFFile(folder / "data.xdmf", [u, v], backend=backend)
         u.interpolate(dolfinx.fem.Expression(u_expr, V.element.interpolation_points()))
         v.interpolate(dolfinx.fem.Expression(v_expr, V.element.interpolation_points()))
-        xdmf.write(t.value)
+        xdmf.write(rtype(t.value))
 
         t.value = 0.3
         u.interpolate(dolfinx.fem.Expression(u_expr, V.element.interpolation_points()))
         v.interpolate(dolfinx.fem.Expression(v_expr, V.element.interpolation_points()))
-        xdmf.write(t.value)
+        xdmf.write(rtype(t.value))
 
         xdmf.close()
 
@@ -103,9 +108,9 @@ def test_XDMFFile_3D(cell_type, degree, value_shape, backend, use_ctx_manager, t
         scheme="default", degree=degree, cell=mesh.ufl_cell().cellname(), value_shape=value_shape
     )
     V = dolfinx.fem.functionspace(mesh, el)
-    u = dolfinx.fem.Function(V)
-    v = dolfinx.fem.Function(V)
-    t = dolfinx.fem.Constant(mesh, 0.0)
+    u = dolfinx.fem.Function(V, dtype=stype)
+    v = dolfinx.fem.Function(V, dtype=stype)
+    t = dolfinx.fem.Constant(mesh, stype(0.0))
 
     X = ufl.SpatialCoordinate(mesh)
     if value_shape == ():
@@ -123,22 +128,22 @@ def test_XDMFFile_3D(cell_type, degree, value_shape, backend, use_ctx_manager, t
         with scifem.xdmf.XDMFFile(folder / "data.xdmf", [u, v], backend=backend) as xdmf:
             u.interpolate(dolfinx.fem.Expression(u_expr, V.element.interpolation_points()))
             v.interpolate(dolfinx.fem.Expression(v_expr, V.element.interpolation_points()))
-            xdmf.write(t.value)
+            xdmf.write(rtype(t.value))
 
             t.value = 0.3
             u.interpolate(dolfinx.fem.Expression(u_expr, V.element.interpolation_points()))
             v.interpolate(dolfinx.fem.Expression(v_expr, V.element.interpolation_points()))
-            xdmf.write(t.value)
+            xdmf.write(rtype(t.value))
     else:
         xdmf = scifem.xdmf.XDMFFile(folder / "data.xdmf", [u, v], backend=backend)
         u.interpolate(dolfinx.fem.Expression(u_expr, V.element.interpolation_points()))
         v.interpolate(dolfinx.fem.Expression(v_expr, V.element.interpolation_points()))
-        xdmf.write(t.value)
+        xdmf.write(rtype(t.value))
 
         t.value = 0.3
         u.interpolate(dolfinx.fem.Expression(u_expr, V.element.interpolation_points()))
         v.interpolate(dolfinx.fem.Expression(v_expr, V.element.interpolation_points()))
-        xdmf.write(t.value)
+        xdmf.write(rtype(t.value))
 
         xdmf.close()