diff --git a/openmc/material.py b/openmc/material.py index 957e0ba2d4..f4c9787981 100644 --- a/openmc/material.py +++ b/openmc/material.py @@ -833,7 +833,7 @@ class Material(IDManagerMixin): ---------- nuclides : str, optional Nuclide for which atom density is desired. If not specified, the - atom density for the all the material is given. + atom density for each nuclide in the material is given. .. versionchanged:: 0.13.1 The values in the dictionary were changed from a tuple containing @@ -868,10 +868,10 @@ class Material(IDManagerMixin): nuc_densities = [] nuc_density_types = [] - for loop_nuclide in self.nuclides: - nucs.append(loop_nuclide.name) - nuc_densities.append(loop_nuclide.percent) - nuc_density_types.append(loop_nuclide.percent_type) + for nuc in self.nuclides: + nucs.append(nuc.name) + nuc_densities.append(nuc.percent) + nuc_density_types.append(nuc.percent_type) nucs = np.array(nucs) nuc_densities = np.array(nuc_densities) diff --git a/tests/unit_tests/test_material.py b/tests/unit_tests/test_material.py index 9425da44fb..8bb8817958 100644 --- a/tests/unit_tests/test_material.py +++ b/tests/unit_tests/test_material.py @@ -339,7 +339,7 @@ def test_get_nuclide_atom_densities(uo2): assert density > 0 -def test_get_specific_nuclide_atom_densities(uo2): +def test_get_nuclide_atom_densities_specific(uo2): one_nuc = uo2.get_nuclide_atom_densities(nuclide='O16') assert list(one_nuc.keys()) == ['O16'] assert list(one_nuc.values())[0] > 0