@@ -273,15 +273,18 @@ def test_momentum_correction_workflow(features: np.ndarray):
273
273
)
274
274
assert len (processor .mc .pouter_ord ) == rotsym
275
275
processor .generate_splinewarp (use_center = include_center )
276
- processor .save_splinewarp (filename = f"sed_config{ len (features )} .yaml" , overwrite = True )
276
+ processor .save_splinewarp (
277
+ filename = f"sed_config_momentum_correction{ len (features )} .yaml" ,
278
+ overwrite = True ,
279
+ )
277
280
pouter_ord = processor .mc .pouter_ord
278
281
cdeform_field = processor .mc .cdeform_field
279
282
rdeform_field = processor .mc .rdeform_field
280
283
# load features
281
284
processor = SedProcessor (
282
285
folder = df_folder ,
283
286
config = config ,
284
- folder_config = f"sed_config { len (features )} .yaml" ,
287
+ folder_config = f"sed_config_momentum_correction { len (features )} .yaml" ,
285
288
user_config = {},
286
289
system_config = {},
287
290
)
@@ -290,7 +293,7 @@ def test_momentum_correction_workflow(features: np.ndarray):
290
293
np .testing .assert_allclose (processor .mc .pouter_ord , pouter_ord )
291
294
np .testing .assert_allclose (processor .mc .cdeform_field , cdeform_field )
292
295
np .testing .assert_allclose (processor .mc .rdeform_field , rdeform_field )
293
- os .remove (f"sed_config { len (features )} .yaml" )
296
+ os .remove (f"sed_config_momentum_correction { len (features )} .yaml" )
294
297
295
298
296
299
def test_pose_adjustment ():
@@ -378,10 +381,11 @@ def test_momentum_calibration_workflow():
378
381
apply = True ,
379
382
)
380
383
assert processor .mc .calibration ["kx_scale" ] != processor .mc .calibration ["ky_scale" ]
381
- processor .save_momentum_calibration ()
384
+ processor .save_momentum_calibration (filename = "sed_config_momentum_calibration.yaml" )
382
385
processor = SedProcessor (
383
386
folder = df_folder ,
384
387
config = config ,
388
+ folder_config = "sed_config_momentum_calibration.yaml" ,
385
389
user_config = {},
386
390
system_config = {},
387
391
)
@@ -392,7 +396,7 @@ def test_momentum_calibration_workflow():
392
396
)
393
397
assert "kx" in processor .dataframe .columns
394
398
assert "ky" in processor .dataframe .columns
395
- os .remove ("sed_config .yaml" )
399
+ os .remove ("sed_config_momentum_calibration .yaml" )
396
400
397
401
398
402
def test_energy_correction ():
@@ -521,19 +525,22 @@ def test_energy_calibration_workflow(energy_scale: str, calibration_method: str)
521
525
method = calibration_method ,
522
526
)
523
527
assert processor .ec .calibration ["energy_scale" ] == energy_scale
524
- processor .save_energy_calibration ()
528
+ processor .save_energy_calibration (
529
+ filename = f"sed_config_energy_calibration_{ energy_scale } -{ calibration_method } .yaml" ,
530
+ )
525
531
processor .append_energy_axis ()
526
532
assert "energy" in processor .dataframe .columns
527
533
processor = SedProcessor (
528
534
folder = df_folder + "../mpes/" ,
529
535
config = config ,
536
+ folder_config = f"sed_config_energy_calibration_{ energy_scale } -{ calibration_method } .yaml" ,
530
537
user_config = {},
531
538
system_config = {},
532
539
)
533
540
processor .append_energy_axis (preview = True )
534
541
assert "energy" in processor .dataframe .columns
535
542
assert processor .attributes ["energy_calibration" ]["calibration" ]["energy_scale" ] == energy_scale
536
- os .remove ("sed_config .yaml" )
543
+ os .remove (f"sed_config_energy_calibration_ { energy_scale } - { calibration_method } .yaml" )
537
544
538
545
539
546
def test_delay_calibration_workflow ():
0 commit comments