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

Fixed Python 3 issue in tally slice-merge test

Browse source
This commit is contained in:
wbinventor@gmail.com 2016-02-11 16:27:03 -05:00
parent 23353fa097
commit baad8a998a
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_tally_slice_merge/test_tally_slice_merge.py
View file

@ -110,6 +110,7 @@ class TallySliceMergeTestHarness(PyAPITestHarness):
# Slice the tallies by score
score_prod = itertools.product(tallies, self.scores)
tallies = map(lambda ts: ts[0].get_slice(scores=[ts[1]]), score_prod)
tallies = list(tallies)
# Initialize an output string
outstr = ''
@ -120,11 +121,10 @@ class TallySliceMergeTestHarness(PyAPITestHarness):
outstr += df.to_string()
# Merge all tallies together
while len(list(tallies)) != 1:
tallies = list(tallies)
while len(tallies) != 1:
halfway = int(len(tallies) / 2)
zip_split = zip(tallies[:halfway], tallies[halfway:])
tallies = map(lambda xy: xy[0].merge(xy[1]), zip_split)
tallies = list(map(lambda xy: xy[0].merge(xy[1]), zip_split))
# Append merged Tally Pandas DataFrame to output string
df = tallies[0].get_pandas_dataframe()