crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 22:26:08 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

revert to original expected ww lb

Browse source
This commit is contained in:
Ethan Peterson 2022-10-03 07:17:07 -04:00
parent 1a306c4c15
commit 4957ba240b
1 changed files with 1 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/unit_tests/weightwindows/test_wwinp_reader.py
View file

@ -118,12 +118,7 @@ def test_wwinp_reader(wwinp_data, request):
# check the expected weight window values mocked in the file --
# a reversed array of the flat index into the numpy array
n_wws = np.prod((*mesh.dimension, e_bounds.size - 1))
nx, ny, nz, ne = mesh.dimension + (e_bounds.size - 1,)
exp_ww_lb = np.linspace(1, n_wws, n_wws)[::-1].reshape(nx, ny, nz, ne)
# this is how the ww lower bounds were actually written to file
exp_ww_lb = np.swapaxes(exp_ww_lb, 0, 2).flatten()
# this is how we expect them to be read back in
exp_ww_lb = exp_ww_lb.reshape(ne, nz, ny, nx).T.flatten()
exp_ww_lb = np.linspace(1, n_wws, n_wws)[::-1]
np.testing.assert_array_equal(exp_ww_lb, ww.lower_ww_bounds.flatten())