use get_reshaped_data in tests

This commit is contained in:
RemDelaporteMathurin 2023-03-07 14:36:07 -05:00
parent 1e94a0c7d7
commit ab4237c037
2 changed files with 2 additions and 2 deletions

View file

@ -93,6 +93,6 @@ def test_offset_mesh(model, estimator, origin):
# check that the lower half of the mesh contains a tally result
# and that the upper half is zero
mean = tally.mean.reshape(mesh.dimension[::-1]).T
mean = tally.get_reshaped_data('mean', expand_dims=True)
# assert np.count_nonzero(mean[:, :, :5]) == mean.size / 2
# assert np.count_nonzero(mean[:, :, 5:]) == 0

View file

@ -93,6 +93,6 @@ def test_offset_mesh(model, estimator, origin):
# check that the lower half of the mesh contains a tally result
# and that the upper half is zero
mean = tally.mean.reshape(mesh.dimension[::-1]).T
mean = tally.get_reshaped_data('mean', expand_dims=True)
# assert np.count_nonzero(mean[:, :, :5]) == mean.size / 2
# assert np.count_nonzero(mean[:, :, 5:]) == 0