diff --git a/tests/accuracy/test_global_correction.py b/tests/accuracy/test_global_correction.py index 623918470..0859a22a6 100644 --- 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 index a29c47219..e3dcd95c6 100644 --- 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 ###############