Add unit test for component flux calc.

terrapower · Jan 31, 2025 · 1f10f53 · 1f10f53
1 parent 04846e7
commit 1f10f53
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/armi/reactor/tests/test_blocks.py b/armi/reactor/tests/test_blocks.py
@@ -1201,6 +1201,20 @@ def test_adjustDensity(self):
 
         self.assertAlmostEqual(mass2 - mass1, massDiff)
 
+    @patch.object(blocks.HexBlock, "getSymmetryFactor")
+    def test_getMgFlux(self, mock_sf):
+        mock_sf.return_value = 3
+        self.block.p.mgFlux = np.array([1, 1, 1, 1, 1])
+        self.block.p.mgFluxGamma = np.array([2, 2, 2, 2])
+        fuel = self.block.getComponent(Flags.FUEL)
+        blockVol = self.block.getVolume()
+        fuelVol = fuel.getVolume()
+        volFrac = fuelVol / blockVol / self.block.getSymmetryFactor()
+        neutronFlux = fuel.getIntegratedMgFlux()
+        gammaFlux = fuel.getIntegratedMgFlux(gamma=True)
+        np.testing.assert_almost_equal(neutronFlux, np.ones(5) * volFrac)
+        np.testing.assert_almost_equal(gammaFlux, np.ones(4) * volFrac * 2.0)
+
     @patch.object(blocks.HexBlock, "getSymmetryFactor")
     def test_completeInitialLoading(self, mock_sf):
         """Ensure that some BOL block and component params are populated properly.