Cleaning out unnecessary uses of r.o (#1999)

Author: john-science

armi/mpiActions.py

@@ -505,7 +505,7 @@ def __init__(self, skipInterfaces=False):
         self._skipInterfaces = skipInterfaces
 
     def invokeHook(self):
-        r"""Sync up all nodes with the reactor, the cs, and the interfaces.
+        """Sync up all nodes with the reactor, the cs, and the interfaces.
 
         Notes
         -----
@@ -604,9 +604,7 @@ def _distributeReactor(self, cs):
 
         self.r.o = self.o
 
-        runLog.debug(
-            "The reactor has {} assemblies".format(len(self.r.core.getAssemblies()))
-        )
+        runLog.debug(f"The reactor has {len(self.r.core.getAssemblies())} assemblies")
         # attach here so any interface actions use a properly-setup reactor.
         self.o.reattach(self.r, cs)  # sets r and cs
 

armi/physics/fuelCycle/tests/test_fuelHandlers.py

@@ -481,7 +481,7 @@ def test_repeatShuffles(self):
             self.assertEqual(a.getLocation(), "SFP")
 
         # do some shuffles
-        fh = self.r.o.getInterface("fuelHandler")
+        fh = self.o.getInterface("fuelHandler")
         self.runShuffling(fh)  # changes caseTitle
 
         # Make sure the generated shuffles file matches the tracked one.  This will need to be
@@ -504,7 +504,7 @@ def test_repeatShuffles(self):
         newSettings["explicitRepeatShuffles"] = "armiRun-SHUFFLES.txt"
         self.o.cs = self.o.cs.modified(newSettings=newSettings)
 
-        fh = self.r.o.getInterface("fuelHandler")
+        fh = self.o.getInterface("fuelHandler")
 
         self.runShuffling(fh)
 

armi/physics/neutronics/macroXSGenerationInterface.py

@@ -25,8 +25,8 @@
 from armi import mpiActions
 from armi import runLog
 from armi.nuclearDataIO import xsCollections
-from armi.utils import iterables
 from armi.physics.neutronics.settings import CONF_MINIMUM_NUCLIDE_DENSITY
+from armi.utils import getBurnSteps, iterables
 
 
 class MacroXSGenerator(mpiActions.MpiAction):
@@ -112,12 +112,12 @@ def invokeHook(self):
 
 class MacroXSGenerationInterface(interfaces.Interface):
     """
-    Builds macroscopic cross sections on all blocks.
-
-    .. warning::
-         This probably shouldn't be an interface since it has no interactXYZ methods
-         It should probably be converted to an MpiAction.
+    Builds macroscopic cross sections on all Blocks.
 
+    Warning
+    -------
+    This probably shouldn't be an interface since it has no interactXYZ methods. It should probably
+    be converted to an MpiAction.
     """
 
     name = "macroXsGen"
@@ -136,8 +136,7 @@ def buildMacros(
         libType="micros",
     ):
         """
-        Builds block-level macroscopic cross sections for making diffusion
-        equation matrices.
+        Builds block-level macroscopic cross sections for making diffusion equation matrices.
 
         This will use MPI if armi.context.MPI_SIZE > 1
 
@@ -188,8 +187,9 @@ def buildMacros(
             block: either "micros" for neutron XS or "gammaXS" for gamma XS.
         """
         cycle = self.r.p.cycle
+        burnSteps = getBurnSteps(self.cs)
         self.macrosLastBuiltAt = (
-            sum([self.r.o.burnSteps[i] + 1 for i in range(cycle)]) + self.r.p.timeNode
+            sum([burnSteps[i] + 1 for i in range(cycle)]) + self.r.p.timeNode
         )
 
         runLog.important("Building macro XS")
@@ -205,10 +205,8 @@ def buildMacros(
         xsGen.invoke(self.o, self.r, self.cs)
 
 
-# helper functions for mpi communication
-
-
 def _scatterList(lst):
+    """Helper functions for mpi communication."""
     if context.MPI_RANK == 0:
         chunked = iterables.split(lst, context.MPI_SIZE)
     else:
@@ -217,6 +215,7 @@ def _scatterList(lst):
 
 
 def _gatherList(localList):
+    """Helper functions for mpi communication."""
     globalList = context.MPI_COMM.gather(localList, root=0)
     if context.MPI_RANK == 0:
         globalList = iterables.flatten(globalList)

armi/reactor/assemblies.py

@@ -1018,28 +1018,24 @@ def getParamValuesAtZ(
         This caches interpolators for each param and must be cleared if new params are
         set or new heights are set.
 
-        WARNING:
-        Fails when requested to extrapolate.With higher order splines it is possible
-        to interpolate non-physical values, for example a negative flux or dpa. Please
-        use caution when going off default in interpType and be certain that
-        interpolated values are physical.
+        Warning
+        -------
+        Fails when requested to extrapolate. With higher order splines it is possible to interpolate
+        non-physical values, for example, a negative flux or dpa. Please use caution when going off
+        default in interpType and be certain that interpolated values are physical.
 
         Parameters
         ----------
         param : str
             the parameter to interpolate
-
         elevations : array of float
-            the elevations from the bottom of the assembly in cm at which you want the
-            point.
-
+            the elevations from the bottom of the assembly in cm at which you want the point.
         interpType: str or int
             used in interp1d. interp1d documention: Specifies the kind of interpolation
             as a string ('linear', 'nearest', 'zero', 'slinear', 'quadratic', 'cubic'
             where 'slinear', 'quadratic' and 'cubic' refer to a spline interpolation of
             first, second or third order) or as an integer specifying the order of the
             spline interpolator to use. Default is 'linear'.
-
         fillValue: str
             Rough pass through to scipy.interpolate.interp1d. If 'extend', then the
             lower and upper bounds are used as the extended value. If 'extrapolate',
@@ -1070,23 +1066,22 @@ def getParamOfZFunction(self, param, interpType="linear", fillValue=np.NaN):
         This caches interpolators for each param and must be cleared if new params are
         set or new heights are set.
 
-        WARNING: Fails when requested to extrapololate. With higher order splines it is
-        possible to interpolate nonphysical values, for example a negative flux or dpa.
-        Please use caution when going off default in interpType and be certain that
-        interpolated values are physical.
+        Warning
+        -------
+        Fails when requested to extrapololate. With higher order splines it is possible to
+        interpolate nonphysical values, for example, a negative flux or dpa. Please use caution when
+        going off default in interpType and be certain that interpolated values are physical.
 
         Parameters
         ----------
         param : str
             the parameter to interpolate
-
         interpType: str or int
             used in interp1d. interp1d documention: Specifies the kind of interpolation
             as a string ('linear', 'nearest', 'zero', 'slinear', 'quadratic', 'cubic'
             where 'slinear', 'quadratic' and 'cubic' refer to a spline interpolation of
             first, second or third order) or as an integer specifying the order of the
             spline interpolator to use. Default is 'linear'.
-
         fillValue: float
             Rough pass through to scipy.interpolate.interp1d. If 'extend', then the
             lower and upper bounds are used as the extended value. If 'extrapolate',

armi/reactor/converters/parameterSweeps/generalParameterSweepConverters.py

@@ -52,15 +52,6 @@ def convert(self, r=None):
             )
 
 
-class CustomModifier(ParameterSweepConverter):
-    """Invoke the shuffle logic `applyCustomPerturbation` method to make a custom setting."""
-
-    def convert(self, r=None):
-        ParameterSweepConverter.convert(self, r)
-        fh = r.o.getInterface("fuelHandler")
-        fh.applyCustomPerturbation(self._parameter)
-
-
 class NeutronicConvergenceModifier(ParameterSweepConverter):
     """Adjusts the neutronics convergence parameters."""
 

armi/reactor/converters/parameterSweeps/tests/test_paramSweepConverters.py

@@ -18,7 +18,6 @@
 
 from armi.tests import TEST_ROOT
 from armi.reactor.converters.parameterSweeps.generalParameterSweepConverters import (
-    CustomModifier,
     NeutronicConvergenceModifier,
     ParameterSweepConverter,
     SettingsModifier,
@@ -64,15 +63,3 @@ def test_settingsModifier(self):
         # NOTE: Settings objects are not modified, but we point to new objects
         self.assertIn("Simple test input", self.cs["comment"])
         self.assertEqual(con._cs["comment"], "FakeParam")
-
-    def test_customModifier(self):
-        """Super basic test of the Custom Modifier."""
-        con = CustomModifier(self.cs, "FakeParam")
-        self.assertEqual(con._parameter, "FakeParam")
-
-        # For testing purposes, we need to add a dummy perturbation
-        fh = self.r.o.getInterface("fuelHandler")
-        fh.applyCustomPerturbation = lambda val: val
-
-        con.convert(self.r)
-        self.assertEqual(con._sourceReactor, self.r)