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
|
from mpi4py import MPI
import numpy as np
import pytest
from packaging.version import parse as _v
import io4dolfinx
@pytest.mark.parametrize("comm", [MPI.COMM_SELF, MPI.COMM_WORLD])
def test_read_write_attributes(comm, backend, tmp_path):
if backend == "adios2":
import adios2
if _v(np.__version__) >= _v("2.0.0") and _v(adios2.__version__) < _v("2.10.2"):
pytest.skip(reason="Cannot use numpy>=2.0.0 and adios2<2.10.2")
attributes1 = {
"a": np.array([1, 2, 3], dtype=np.uint8),
"b": np.array([4, 5], dtype=np.uint8),
}
attributes2 = {
"c": np.array([6], dtype=np.uint8),
"d": np.array([7, 8, 9, 10], dtype=np.float64),
}
fname = comm.bcast(tmp_path, root=0)
fname = fname / "attributes"
suffix = ".bp" if backend == "adios2" else ".h5"
file = fname.with_suffix(suffix)
# print(comm.size)
io4dolfinx.write_attributes(
comm=comm, filename=file, name="group1", attributes=attributes1, backend=backend
)
io4dolfinx.write_attributes(
comm=comm, filename=file, name="group2", attributes=attributes2, backend=backend
)
loaded_attributes1 = io4dolfinx.read_attributes(
comm=comm, filename=file, name="group1", backend=backend
)
loaded_attributes2 = io4dolfinx.read_attributes(
comm=comm, filename=file, name="group2", backend=backend
)
for k, v in loaded_attributes1.items():
assert np.allclose(v, attributes1[k])
for k, v in attributes1.items():
assert np.allclose(v, loaded_attributes1[k])
for k, v in loaded_attributes2.items():
assert np.allclose(v, attributes2[k])
for k, v in attributes2.items():
assert np.allclose(v, loaded_attributes2[k])
|
