From c036403c984067946542e97cbdbe488687b6b706 Mon Sep 17 00:00:00 2001 From: Jonathan Shimwell Date: Thu, 4 Aug 2022 14:55:45 +0100 Subject: [PATCH] simpler if else statement for normalization Co-authored-by: Patrick Shriwise --- openmc/material.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/openmc/material.py b/openmc/material.py index 48684ed67f..965c9d5c85 100644 --- a/openmc/material.py +++ b/openmc/material.py @@ -929,8 +929,10 @@ class Material(IDManagerMixin): if normalization == 'total': multiplier = self.volume - else: - multiplier = 1 if normalization == 'volume' else 1.0 / self.get_mass_density() + elif normalization == 'volume': + multiplier = 1 + elif normalization == 'mass': + multiplier = 1.0 / self.get_mass_density() activity = {} for nuclide, atoms_per_bcm in self.get_nuclide_atom_densities().items():