bec/ophyd_devices
0
0
Fork 0
mirror of https://github.com/bec-project/ophyd_devices.git synced 2025-06-01 01:20:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: bugfix tests DDG

Browse Source
This commit is contained in:
appel_c 2023-11-23 11:22:17 +01:00
parent 0dda7300e5
commit 9e67a7a7d4
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tests/test_delay_generator_csaxs.py
View File

@ -30,7 +30,7 @@ def mock_DDGSetup():
"scan_type": "step", "scan_type": "step",
"num_points": 500, "num_points": 500,
"frames_per_trigger": 1, "frames_per_trigger": 1,
"exposure_time": 0.1, "exp_time": 0.1,
"readout_time": 0.1, "readout_time": 0.1,
}, },
{ {
@ -38,7 +38,7 @@ def mock_DDGSetup():
"scan_type": "step", "scan_type": "step",
"num_points": 500, "num_points": 500,
"frames_per_trigger": 5, "frames_per_trigger": 5,
"exposure_time": 0.01, "exp_time": 0.01,
"readout_time": 0, "readout_time": 0,
}, },
{ {
@ -46,7 +46,7 @@ def mock_DDGSetup():
"scan_type": "fly", "scan_type": "fly",
"num_points": 500, "num_points": 500,
"frames_per_trigger": 1, "frames_per_trigger": 1,
"exposure_time": 1, "exp_time": 1,
"readout_time": 0.2, "readout_time": 0.2,
}, },
{ {
@ -54,7 +54,7 @@ def mock_DDGSetup():
"scan_type": "fly", "scan_type": "fly",
"num_points": 500, "num_points": 500,
"frames_per_trigger": 5, "frames_per_trigger": 5,
"exposure_time": 0.1, "exp_time": 0.1,
"readout_time": 0.4, "readout_time": 0.4,
}, },
] ]
@ -104,7 +104,7 @@ def ddg_config_defaults(request):
{ {
"fixed_ttl_width": [0, 0, 0, 0, 0], "fixed_ttl_width": [0, 0, 0, 0, 0],
"trigger_width": 0.1, "trigger_width": 0.1,
"set_high_on_exposure": False, "set_high_on_exposure": True,
"set_high_on_stage": False, "set_high_on_stage": False,
"set_trigger_source": "SINGLE_SHOT", "set_trigger_source": "SINGLE_SHOT",
"premove_trigger": True, "premove_trigger": True,
@ -256,12 +256,12 @@ def test_prepare_ddg(
if ddg_config_scan["set_high_on_exposure"]: if ddg_config_scan["set_high_on_exposure"]:
num_burst_cycle = 1 + ddg_config_defaults["additional_triggers"] num_burst_cycle = 1 + ddg_config_defaults["additional_triggers"]
exp_time = ddg_config_defaults["delta_width"] + scaninfo["frames_per_trigger"] * ( exp_time = ddg_config_defaults["delta_width"] + scaninfo["frames_per_trigger"] * (
scaninfo["exposure_time"] + scaninfo["readout_time"] scaninfo["exp_time"] + scaninfo["readout_time"]
) )
total_exposure = exp_time total_exposure = exp_time
delay_burst = ddg_config_defaults["delay_burst"] delay_burst = ddg_config_defaults["delay_burst"]
else: else:
exp_time = ddg_config_defaults["delta_width"] + scaninfo["exposure_time"] exp_time = ddg_config_defaults["delta_width"] + scaninfo["exp_time"]
total_exposure = exp_time + scaninfo["readout_time"] total_exposure = exp_time + scaninfo["readout_time"]
delay_burst = ddg_config_defaults["delay_burst"] delay_burst = ddg_config_defaults["delay_burst"]
num_burst_cycle = ( num_burst_cycle = (
@ -272,19 +272,19 @@ def test_prepare_ddg(
num_burst_cycle = 1 + ddg_config_defaults["additional_triggers"] num_burst_cycle = 1 + ddg_config_defaults["additional_triggers"]
exp_time = ( exp_time = (
ddg_config_defaults["delta_width"] ddg_config_defaults["delta_width"]
+ scaninfo["num_points"] * scaninfo["exposure_time"] + scaninfo["num_points"] * scaninfo["exp_time"]
+ (scaninfo["num_points"] - 1) * scaninfo["readout_time"] + (scaninfo["num_points"] - 1) * scaninfo["readout_time"]
) )
total_exposure = exp_time total_exposure = exp_time
delay_burst = ddg_config_defaults["delay_burst"] delay_burst = ddg_config_defaults["delay_burst"]
else: else:
exp_time = ddg_config_defaults["delta_width"] + scaninfo["exposure_time"] exp_time = ddg_config_defaults["delta_width"] + scaninfo["exp_time"]
total_exposure = exp_time + scaninfo["readout_time"] total_exposure = exp_time + scaninfo["readout_time"]
delay_burst = ddg_config_defaults["delay_burst"] delay_burst = ddg_config_defaults["delay_burst"]
num_burst_cycle = scaninfo["num_points"] + ddg_config_defaults["additional_triggers"] num_burst_cycle = scaninfo["num_points"] + ddg_config_defaults["additional_triggers"]
assert mock_DDGSetup.parent.burst_enable.called_once_with( assert mock_DDGSetup.parent.burst_enable.called_once_with(
num_burst_cycle, delay_burst, total_exposure, config="first" mock.call(num_burst_cycle, delay_burst, total_exposure, config="first")
) )
if not ddg_config_scan["trigger_width"]: if not ddg_config_scan["trigger_width"]:
calls = mock.call("width", exp_time) calls = mock.call("width", exp_time)
@ -295,8 +295,9 @@ def test_prepare_ddg(
if ddg_config_scan["set_high_on_exposure"]: if ddg_config_scan["set_high_on_exposure"]:
calls = [ calls = [
mock.call("width", value, channels=[channel]) mock.call("width", value, channels=[channel])
for channel, value in zip( for value, channel in zip(
ddg_config_defaults["fixed_ttl_width"], channel_pairs["all_channels"] ddg_config_scan["fixed_ttl_width"], channel_pairs["all_channels"]
) )
if value != 0
] ]
assert mock_DDGSetup.parent.set_channels.has_calls(calls) assert mock_DDGSetup.parent.set_channels.has_calls(calls)