1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
import urllib.request
from enum import Enum
from mpi4py import MPI
import numpy as np
import pytest
import io4dolfinx
netcdf4 = pytest.importorskip("netCDF4")
class DownloadStatus(Enum):
success = 1
failed = -1
no_connection = -2
def download_file_if_not_exists(
url, filename, comm: MPI.Intracomm = MPI.COMM_WORLD, rank: int = 0
) -> DownloadStatus:
status = DownloadStatus.failed
if comm.rank == rank:
if not filename.exists():
try:
urllib.request.urlretrieve(url, filename)
status = DownloadStatus.success
except urllib.error.URLError as e:
if str(e) == "<urlopen error [Errno -3] Temporary failure in name resolution>":
status = DownloadStatus.no_connection
else:
status = DownloadStatus.failed
else:
status = DownloadStatus.success
status = comm.bcast(status, root=rank)
comm.Barrier()
return status
def test_read_mesh_and_cell_data(tmp_path):
tmp_path = MPI.COMM_WORLD.bcast(tmp_path, root=0)
filename = tmp_path / "openmc_master_out_openmc0.e"
url = "https://github.com/neams-th-coe/cardinal/blob/devel/test/tests/neutronics/feedback/single_level/gold/openmc_master_out_openmc0.e?raw=true"
status = download_file_if_not_exists(url, filename)
if status == DownloadStatus.no_connection:
pytest.skip("No internet connection")
mesh = io4dolfinx.read_mesh(filename, MPI.COMM_WORLD, backend="exodus")
io4dolfinx.read_meshtags(filename, mesh, meshtag_name="cell", backend="exodus")
io4dolfinx.read_meshtags(filename, mesh, meshtag_name="facet", backend="exodus")
io4dolfinx.read_cell_data(
filename, name="cell_temperature", mesh=mesh, backend="exodus", time=1.0
)
def test_read_mesh_point_data(tmp_path):
tmp_path = MPI.COMM_WORLD.bcast(tmp_path, root=0)
filename = tmp_path / "openmc_master_out_openmc0.e"
url = "https://github.com/idaholab/moose/blob/next/test/tests/kernels/2d_diffusion/gold/matdiffusion_out.e?raw=true"
status = download_file_if_not_exists(url, filename)
if status == DownloadStatus.no_connection:
pytest.skip("No internet connection")
mesh = io4dolfinx.read_mesh(filename, MPI.COMM_WORLD, backend="exodus")
u = io4dolfinx.read_point_data(filename, name="u", mesh=mesh, backend="exodus", time=1.0)
assert mesh.topology.index_map(mesh.topology.dim).size_global == 4
assert mesh.geometry.index_map().size_global == 9
assert np.isclose(mesh.comm.allreduce(np.max(u.x.array), op=MPI.MAX), 1.1140844375981802)
def test_read_second_order_mesh(tmp_path):
tmp_path = MPI.COMM_WORLD.bcast(tmp_path, root=0)
filename = tmp_path / "box-test_out_nek0.e"
url = "https://github.com/neams-th-coe/cardinal/blob/devel/test/tests/deformation/simple-cube/gold/box-test_out_nek0.e?raw=true"
status = download_file_if_not_exists(url, filename)
if status == DownloadStatus.no_connection:
pytest.skip("No internet connection")
mesh = io4dolfinx.read_mesh(filename, MPI.COMM_WORLD, backend="exodus", time=5)
assert mesh.topology.index_map(mesh.topology.dim).size_global == 64
assert mesh.geometry.index_map().size_global == 1728
u_x = io4dolfinx.read_point_data(filename, name="disp_x", mesh=mesh, backend="exodus", time=5.0)
ref_min = -0.7699306351843485
ref_max = 0.7699306351843485
assert np.isclose(mesh.comm.allreduce(np.min(u_x.x.array), op=MPI.MIN), ref_min)
assert np.isclose(mesh.comm.allreduce(np.max(u_x.x.array), op=MPI.MAX), ref_max)
|
