crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 05:35:49 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test of volume normalization ordering

Browse source
This commit is contained in:
Patrick Shriwise 2023-02-23 23:59:46 -06:00
parent a270d4c01a
commit 70b743e5ed
1 changed files with 12 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/unit_tests/test_mesh_to_vtk.py
View file

@ -34,7 +34,11 @@ def test_write_data_to_vtk(mesh, tmpdir):
# BUILD
filename = Path(tmpdir) / "out.vtk"
data = np.random.random(mesh.num_mesh_cells)
# use mesh element volumes as data to check volume-normalization ordering
# kji (i changing fastest) orering is expected for input data
# by using the volumes transposed as the data here, we can ensure the
# normalization is happening correctly
data = mesh.volumes.T
# RUN
mesh.write_data_to_vtk(filename=filename, datasets={"label1": data, "label2": data})
@ -56,8 +60,13 @@ def test_write_data_to_vtk(mesh, tmpdir):
assert array2.GetName() == "label2"
# check size of datasets
assert nps.vtk_to_numpy(array1).size == data.size
assert nps.vtk_to_numpy(array2).size == data.size
data1 = nps.vtk_to_numpy(array1)
data2 = nps.vtk_to_numpy(array2)
assert data1.size == data.size
assert data2.size == data.size
assert all(data1 == data2)
assert all(data1 == 1.0)
@pytest.mark.parametrize("mesh", [cylinder_mesh, regular_mesh, rectilinear_mesh, spherical_mesh])