move functionality of updating detailedNDens within changeNDensByFact…

…or and update testing
terrapower · Oct 21, 2024 · 3bb63bf · 3bb63bf
1 parent 16a884f
commit 3bb63bf
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/armi/reactor/composites.py b/armi/reactor/composites.py
@@ -1575,6 +1575,9 @@ def changeNDensByFactor(self, factor):
             nuc: val * factor for nuc, val in self.getNumberDensities().items()
         }
         self.setNumberDensities(densitiesScaled)
+        # Update detailedNDens
+        if self.p.detailedNDens:
+            self.p.detailedNDens *= factor
 
     def clearNumberDensities(self):
         """

diff --git a/armi/reactor/converters/axialExpansionChanger/axialExpansionChanger.py b/armi/reactor/converters/axialExpansionChanger/axialExpansionChanger.py
@@ -343,11 +343,7 @@ def axiallyExpandAssembly(self):
                             c.zbottom = self.linked.linkedBlocks[b][0].p.ztop
                     c.ztop = c.zbottom + c.height
                     # update component number densities
-                    newNumberDensities = {
-                        nuc: c.getNumberDensity(nuc) / growFrac
-                        for nuc in c.getNuclides()
-                    }
-                    c.setNumberDensities(newNumberDensities)
+                    c.changeNDensByFactor(1.0 / growFrac)
                     # redistribute block boundaries if on the target component
                     if self.expansionData.isTargetComponent(c):
                         b.p.ztop = c.ztop

diff --git a/armi/reactor/tests/test_components.py b/armi/reactor/tests/test_components.py
@@ -697,9 +697,11 @@ def test_getNumberDensities(self):
     def test_changeNumberDensities(self):
         """Test that demonstates that the number densities on a component can be modified."""
         self.component.p.numberDensities = {"NA23": 1.0}
+        self.component.p.detailedNDens = [1.0]
         self.assertEqual(self.component.getNumberDensity("NA23"), 1.0)
         self.component.changeNDensByFactor(3.0)
         self.assertEqual(self.component.getNumberDensity("NA23"), 3.0)
+        self.assertEqual(self.component.p.detailedNDens[0], 3.0)
 
     def test_fuelMass(self):
         nominalMass = self.component.getMass()