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

improved test fixed sum bug

Browse source
This commit is contained in:
Jonathan Shimwell 2022-08-23 16:42:22 +01:00
parent 085125ef86
commit 6171e09ed3
2 changed files with 12 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

10
openmc/material.py
View file

@ -869,10 +869,9 @@ class Material(IDManagerMixin):
nuc_density_types = []
for loop_nuclide in self.nuclides:
if nuclide is None or nuclide == loop_nuclide.name:
nucs.append(loop_nuclide.name)
nuc_densities.append(loop_nuclide.percent)
nuc_density_types.append(loop_nuclide.percent_type)
nucs.append(loop_nuclide.name)
nuc_densities.append(loop_nuclide.percent)
nuc_density_types.append(loop_nuclide.percent_type)
nucs = np.array(nucs)
nuc_densities = np.array(nuc_densities)
@ -904,7 +903,8 @@ class Material(IDManagerMixin):
nuclides = OrderedDict()
for n, nuc in enumerate(nucs):
nuclides[nuc] = nuc_densities[n]
if nuclide is None or nuclide == nuc:
nuclides[nuc] = nuc_densities[n]
return nuclides

10
tests/unit_tests/test_material.py
View file

@ -341,9 +341,13 @@ def test_get_nuclide_atom_densities(uo2):
def test_get_specific_nuclide_atom_densities(uo2):
print(uo2)
nuc = uo2.get_nuclide_atom_densities(nuclide='O16')
assert list(nuc.keys()) == ['O16']
assert list(nuc.values())[0] > 0
one_nuc = uo2.get_nuclide_atom_densities(nuclide='O16')
assert list(one_nuc.keys()) == ['O16']
assert list(one_nuc.values())[0] > 0
all_nuc = uo2.get_nuclide_atom_densities()
assert all_nuc['O16'] == one_nuc['O16']
def test_get_nuclide_atoms():