adapt test to new category name

pylhc · Aug 28, 2023 · 0c27d6e · 0c27d6e
1 parent 31d67c5
commit 0c27d6e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/accuracy/test_global_correction.py b/tests/accuracy/test_global_correction.py
@@ -74,7 +74,7 @@ def get_normal_params(beam):
         correction_filename=CORRECTION_TEST_INPUTS / f"changeparameters_injb{beam}_quadrupole.madx",
         optics_params=[f"{PHASE}X", f"{PHASE}Y", f"{BETA}X", f"{BETA}Y", f"{NORM_DISPERSION}X", TUNE],
         weights=[1., 1., 1., 1., 1., 1.],
-        variables=["MQY"],
+        variables=["MQY_Q4"],
         fullresponse="fullresponse_MQY.h5",
         seed=12368,  # iteration test might not work with other seeds (converges too fast)
     )

diff --git a/tests/accuracy/test_response_creator.py b/tests/accuracy/test_response_creator.py
@@ -45,7 +45,7 @@ def test_response_accuracy(model_inj_beams, orientation, creator):
     # compare to original response matrix
     original_response = read_fullresponse(model_inj_beams.model_dir / correction_params.fullresponse)
     for key in optics_params:
-        original = original_response[key]
+        original = original_response[key.replace("_Q4", "")]  # renaming of category since response creation 
         new = new_response[key].loc[original.index, original.columns]
 
         # ######## Relative RMS check ###############