fix various typos and bugs causing tests to fail

Author: nglaser3

mcdc/transport/physics/interface.py

@@ -51,14 +51,12 @@ def total_xs(particle_container, simulation, data):
     """
     particle = particle_container[0]
     if particle["particle_type"] == PARTICLE_NEUTRON:
-        module = neutron
-        type_total = NEUTRON_REACTION_TOTAL
+        total = NEUTRON_REACTION_TOTAL
+        return neutron.macro_xs(total, particle_container, simulation, data)
     elif particle["particle_type"] == PARTICLE_ELECTRON:
-        module = electron
-        type_total = ELECTRON_REACTION_TOTAL
-    else:
-        return 0.0
-    return module.macro_xs(type_total, particle_container, simulation, data)
+        total = ELECTRON_REACTION_TOTAL
+        return electron.macro_xs(total, particle_container, simulation, data)
+    return 0.0
 
 
 @njit

mcdc/transport/technique.py

@@ -2,6 +2,7 @@
 import math
 
 from numba import njit
+import numba
 
 ####
 
@@ -139,6 +140,8 @@ def forced_collision_roulette(particle_container, program, data):
 
     # get index into arrays
     index = get_forced_collision_cell_index(particle_container, fc_object, data)
+    if index < 0:
+        return
 
     # get weights
     threshold = mcdc_get.forced_collisions.threshold_weights(index, fc_object, data)
@@ -178,9 +181,7 @@ def get_forced_collision_cell_index(particle_container, fc_object, data):
             return index
 
     # should never hit this, but just to be safe
-    print_error(
-        f"Failed to find {particle['cell_ID']} in list of cell ids for forced collisions"
-    )
+    return -1
 
 
 @njit