Update BetaYields

fuse/plugins/micro_physics/yields.py

@@ -222,9 +222,10 @@ def get_quanta(self, interactions_in_roi):
     def quanta_from_spline(self, energy):
 
         with open(self.beta_quanta_spline, "rb") as f:
-            quanta_function = pickle.load(f)
+            cs1_poly, cs2_poly = pickle.load(f)
 
-        beta_photons, beta_electrons = quanta_function(energy)
+        beta_photons = cs1_poly(energy)
+        beta_electrons = cs2_poly(energy)
 
         if self.use_recombination_fluctuation:
 

tests/test_MicroPhysics_alt_Yields.py

@@ -17,7 +17,7 @@ def yields_dummy_func(x):
     To be used as a dummy function for the BetaYields plugin. Needs to
     be defined outside the test class to be picklable.
     """
-    return 40, 30
+    return 20000
 
 
 class TestAlternativeYields(unittest.TestCase):
@@ -62,7 +62,7 @@ def test_BetaYields(self):
         # as a function of energy
         spline_func_name = os.path.join(self.temp_dir.name, "beta_quanta_spline.pkl")
         with open(spline_func_name, "wb") as f:
-            pickle.dump(yields_dummy_func, f)
+            pickle.dump((yields_dummy_func, yields_dummy_func), f)
 
         self.test_context.set_config({"beta_quanta_spline": spline_func_name})