From eb9dcac3b5f397fc174bcc1d333ba350b4addbf4 Mon Sep 17 00:00:00 2001
From: church89 <l.chierici89@gmail.com>
Date: Thu, 26 Feb 2026 14:33:10 +0100
Subject: [PATCH] update regression tests

---
 .../ref_depletion_with_refuel.h5              | Bin 28640 -> 28640 bytes
 .../ref_depletion_with_rotation.h5            | Bin 28640 -> 28640 bytes
 .../ref_depletion_with_translation.h5         | Bin 28640 -> 28640 bytes
 .../deplete_with_keff_search_control/test.py  |  21 ++++++++++++------
 4 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/tests/regression_tests/deplete_with_keff_search_control/ref_depletion_with_refuel.h5 b/tests/regression_tests/deplete_with_keff_search_control/ref_depletion_with_refuel.h5
index dbdacd19aff69fe278ecb4de2c6e9618f2e30db9..a335e3b47800c98e4e001dfc2159f884a31c6bf0 100644
GIT binary patch
delta 135
zcmaEGpYg$c#tj1b>YEeI?uY*~a4~-#X7G7SuS<Vp-i{3mKe^m-IMQ&N^R2y<U$Qhu
zeA#64d|9413vSvofWZU%i5m?JHbM28q3P96ta8)*{Kj6dE!@Jlxd@_1;;X}ChkOwv
HLl_(YL3uY*

delta 135
zcmaEGpYg$c#tj1b>hBlno(=zJ;IhyDg23l3b6tM#Ua@4u!cQ(YR;Af8OMJBVKUFU}
zi79`wdA=;q`)^O|jltlv{ltw12DtQ!|B+q0((t1_gJa^!^`8sv89+edtHWf6d=aoA
N?_X{FGehIOJpl6dLb?C|

diff --git a/tests/regression_tests/deplete_with_keff_search_control/ref_depletion_with_rotation.h5 b/tests/regression_tests/deplete_with_keff_search_control/ref_depletion_with_rotation.h5
index 820eddcc39654733eeb8bd79636a01b2679a6813..c8b4f67ff25dcfec4cd6d02aeea9443077d6feed 100644
GIT binary patch
delta 91
zcmaEGpYg$c#tj<z{AE5$oFzNo*$0OHiafWvWU_g_EYFE4&+Qq&AmqTrjRun^<V#3A
fJsI=j=gD{WXF`LL><o(`@)BPiCOhPdplAR9@VX;R

delta 91
zcmaEGpYg$c#tj<z{FOWw27mcK*iQ(k{qaM*Y_fU2ERSmab9)9b@H{Yaqrv0}`4SSz
n&oggdTJ_%k6pPvJ9Z$<3@)BPiCOhPdfHmmmnWjE?>U{tJ!CNH>

diff --git a/tests/regression_tests/deplete_with_keff_search_control/ref_depletion_with_translation.h5 b/tests/regression_tests/deplete_with_keff_search_control/ref_depletion_with_translation.h5
index f7f3ff655dcc3ee95e76ac819368d75a5a4ea7c3..35285321bc2a30f35f474f4b123ac68750eb1b0c 100644
GIT binary patch
delta 91
zcmaEGpYg$c#tj<z{1I<w&kPfOZ_mxHf0AiI`DF8aS)P51U)nQ(f!cwI8x1B;$d{0)
fjNEl`fysOO&>JG&-aW+-d5NzMlO6I!P&5DlbrB$;

delta 91
zcmaEGpYg$c#tj<z{2ILzfBUO`uum@xUne14I@vs5mPc~y3;V10dJZfwS3EFrqrv0}
t`4SR4IUmlvefPcnOmqE(&-{w)89+edtHWf6d=apQmTA+MwL~f&003j=C)xl2

diff --git a/tests/regression_tests/deplete_with_keff_search_control/test.py b/tests/regression_tests/deplete_with_keff_search_control/test.py
index aa1ec0942..0af6acd53 100644
--- a/tests/regression_tests/deplete_with_keff_search_control/test.py
+++ b/tests/regression_tests/deplete_with_keff_search_control/test.py
@@ -1,5 +1,6 @@
 """ Tests for KeffSearchControl class """
 
+from hmac import new
 from pathlib import Path
 import shutil
 import sys
@@ -89,14 +90,16 @@ def rotate_cell(angle):
 def adjust_material_density(density_factor):
     f = [material for material in openmc.lib.materials.values() if material.name == 'f'][0]
     nuclides = openmc.lib.materials[f.id].nuclides
-    current_densities = openmc.lib.materials[f.id].densities
-    new_densities = [d * density_factor for d in current_densities]
-    openmc.lib.materials[f.id].set_densities(nuclides, new_densities)
+    densities = openmc.lib.materials[f.id].densities
+    nuc_idx = nuclides.index('U235')
+    new_density = densities[nuc_idx] * density_factor
+    densities[nuc_idx] = new_density
+    openmc.lib.materials[f.id].set_densities(nuclides, densities)
 
 @pytest.mark.parametrize("function, x0, x1, bracket, ref_result", [
     (translate_cell, -11, -5, [-15,0], 'depletion_with_translation'),
     (rotate_cell, -80, -50, [-90,0], 'depletion_with_rotation'),
-    (adjust_material_density, 0.5, 1.2, [0.2, 1.5], 'depletion_with_refuel')
+    (adjust_material_density, 0.5, 2, [0.3, 3.0], 'depletion_with_refuel')
     ])
 
 def test_keff_search_control(run_in_tmpdir, model, function, x0, x1, bracket, ref_result):
@@ -106,10 +109,14 @@ def test_keff_search_control(run_in_tmpdir, model, function, x0, x1, bracket, re
 
     integrator = openmc.deplete.PredictorIntegrator(
         op, [1], 174., timestep_units = 'd')
-    integrator.add_keff_search_control(function, x0, x1, bracket, 
+    integrator.add_keff_search_control(
+        function=function,
+        x0=x0,
+        x1=x1,
+        bracket=bracket,
         output=True, 
-        k_tol=1e-2, 
-        sigma_final=5e-3)
+        k_tol=1e-1, 
+        sigma_final=5e-2)
     
     integrator.integrate()