From 8e90cf504febf507ee20ba62a5173afa9674159a Mon Sep 17 00:00:00 2001
From: Michael Jarrett <mjarrett@terrapower.com>
Date: Tue, 12 Mar 2024 10:58:41 -0700
Subject: [PATCH] Modify mass conservation tests

---
 armi/reactor/tests/test_assemblies.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/armi/reactor/tests/test_assemblies.py b/armi/reactor/tests/test_assemblies.py
index a2435cdfc..572ee2dc0 100644
--- a/armi/reactor/tests/test_assemblies.py
+++ b/armi/reactor/tests/test_assemblies.py
@@ -1151,6 +1151,10 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         ################################
         igniterFuel = self.r.core.childrenByLocator[grid[0, 0, 0]]
         # gridplate, fuel, fuel, fuel, plenum
+        for b in igniterFuel.getBlocks(Flags.FUEL):
+            fuelComp = b.getComponent(Flags.FUEL)
+            fuelComp.setNumberDensity("FE56", 1e-10)
+            fuelComp.setNumberDensity("NA23", 1e-10)
         b = igniterFuel[0]
         coolantNucs = b.getComponent(Flags.COOLANT).getNuclides()
         coolMass = 0
@@ -1163,6 +1167,7 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         igniterHMMass1 = b.getHMMass()
         igniterZircMass1 = b.getMass("ZR")
         igniterFuelBlockMass = b.getMass()
+        igniterDuctMass = b.getComponent(Flags.DUCT).getMass()
 
         coolMass = 0
         b = igniterFuel[4]
@@ -1188,6 +1193,7 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         b = igniterFuel[1]
         igniterHMMass1AfterExpand = b.getHMMass()
         igniterZircMass1AfterExpand = b.getMass("ZR")
+        igniterDuctMassAfterExpand = b.getComponent(Flags.DUCT).getMass()
 
         coolMass = 0
         b = igniterFuel[4]
@@ -1198,6 +1204,9 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         self.assertAlmostEqual(igniterMassGrid, igniterMassGridAfterExpand, 7)
         self.assertAlmostEqual(igniterHMMass1, igniterHMMass1AfterExpand, 7)
         self.assertAlmostEqual(igniterZircMass1, igniterZircMass1AfterExpand, 7)
+        self.assertAlmostEqual(
+            igniterDuctMass, igniterDuctMassAfterExpand * 25.0 / 26.0, 7
+        )
         # Note the masses are linearly different by the amount that the plenum shrunk
         self.assertAlmostEqual(
             igniterPlenumMass, igniterPlenumMassAfterExpand * 75 / 67.0, 7
@@ -1223,6 +1232,7 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         igniterHMMass1AfterShrink = b.getHMMass()
         igniterZircMass1AfterShrink = b.getMass("ZR")
         igniterFuelBlockMassAfterShrink = b.getMass()
+        igniterDuctMassAfterShrink = b.getComponent(Flags.DUCT).getMass()
 
         coolMass = 0
         b = igniterFuel[4]
@@ -1235,6 +1245,7 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         self.assertAlmostEqual(igniterHMMass1, igniterHMMass1AfterShrink, 7)
         self.assertAlmostEqual(igniterZircMass1, igniterZircMass1AfterShrink, 7)
         self.assertAlmostEqual(igniterFuelBlockMass, igniterFuelBlockMassAfterShrink, 7)
+        self.assertAlmostEqual(igniterDuctMass, igniterDuctMassAfterShrink, 7)
         self.assertAlmostEqual(igniterPlenumMass, igniterPlenumMassAfterShrink, 7)
 
     def test_snapAxialMeshToReferenceConservingMassBasedOnBlockShield(self):