Add coolant mass to setBlockMesh test.

terrapower · Mar 12, 2024 · 97215f6 · 97215f6
1 parent 8e90cf5
commit 97215f6
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/armi/reactor/tests/test_assemblies.py b/armi/reactor/tests/test_assemblies.py
@@ -1168,6 +1168,7 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         igniterZircMass1 = b.getMass("ZR")
         igniterFuelBlockMass = b.getMass()
         igniterDuctMass = b.getComponent(Flags.DUCT).getMass()
+        igniterCoolMass = b.getComponent(Flags.COOLANT).getMass()
 
         coolMass = 0
         b = igniterFuel[4]
@@ -1194,6 +1195,7 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         igniterHMMass1AfterExpand = b.getHMMass()
         igniterZircMass1AfterExpand = b.getMass("ZR")
         igniterDuctMassAfterExpand = b.getComponent(Flags.DUCT).getMass()
+        igniterCoolMassAfterExpand = b.getComponent(Flags.COOLANT).getMass()
 
         coolMass = 0
         b = igniterFuel[4]
@@ -1207,6 +1209,9 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         self.assertAlmostEqual(
             igniterDuctMass, igniterDuctMassAfterExpand * 25.0 / 26.0, 7
         )
+        self.assertAlmostEqual(
+            igniterCoolMass, igniterCoolMassAfterExpand * 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
@@ -1233,6 +1238,7 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         igniterZircMass1AfterShrink = b.getMass("ZR")
         igniterFuelBlockMassAfterShrink = b.getMass()
         igniterDuctMassAfterShrink = b.getComponent(Flags.DUCT).getMass()
+        igniterCoolMassAfterShrink = b.getComponent(Flags.COOLANT).getMass()
 
         coolMass = 0
         b = igniterFuel[4]
@@ -1246,6 +1252,7 @@ def test_snapAxialMeshToReferenceConservingMassBasedOnBlockIgniter(self):
         self.assertAlmostEqual(igniterZircMass1, igniterZircMass1AfterShrink, 7)
         self.assertAlmostEqual(igniterFuelBlockMass, igniterFuelBlockMassAfterShrink, 7)
         self.assertAlmostEqual(igniterDuctMass, igniterDuctMassAfterShrink, 7)
+        self.assertAlmostEqual(igniterCoolMass, igniterCoolMassAfterShrink, 7)
         self.assertAlmostEqual(igniterPlenumMass, igniterPlenumMassAfterShrink, 7)
 
     def test_snapAxialMeshToReferenceConservingMassBasedOnBlockShield(self):