def test_get_fpga_temp(mocker):
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 2
m = mocker.patch.object(DetectorApi, 'getAdc', autospec=True)
m.return_value = 34253
d = Eiger()
t = d.temp.fpga[:]
assert t == [34.253, 34.253]
def test_set_vcp_single_call(mocker):
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 2
m = mocker.patch.object(DetectorApi, 'setDac', autospec=True)
# m.return_value = 1560
d = Eiger()
d.dacs.vcp[1] = 1637
m.assert_called_once_with('vcp', 1, 1637)
def test_get_vrf_for_three_mod(mocker):
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 3
m = mocker.patch.object(DetectorApi, 'getDac', autospec=True)
m.return_value = 1560
d = Eiger()
vrf = d.dacs.vrf[:]
assert vrf == [1560, 1560, 1560]
class EigerClientWrapper(object):
def __init__(self):
self.new_client = Eiger()
self.old_client = DetectorClient()
def start(self):
self.new_client.start_detector()
def stop(self):
self.new_client.stop_detector()
def set_config(self, configuration):
self.old_client.set_config(configuration)
def get_status(self):
return self.new_client.status
def set_threshold(self, energy):
self.new_client.threshold = energy
def test_set_eiger_set_from_array_call_count(mocker):
import numpy as np
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 3
m = mocker.patch.object(DetectorApi, 'setDac', autospec=True)
# m.return_value = 1560
d = Eiger()
d.dacs.set_from_array(np.zeros((17, 3)))
assert m.call_count == 17 * 3
def test_iterate_on_index_call_vcn(mocker):
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 10
m = mocker.patch.object(DetectorApi, 'setDac', autospec=True)
# m.return_value = 1560
d = Eiger()
d.dacs.vcn[0, 3, 8] = 1532
calls = [call('vcn', 0, 1532), call('vcn', 3, 1532), call('vcn', 8, 1532)]
m.assert_has_calls(calls)
assert m.call_count == 3
def test_set_vrf_for_three_mod_same_value(mocker):
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 3
m = mocker.patch.object(DetectorApi, 'setDac', autospec=True)
# m.return_value = 1560
d = Eiger()
d.dacs.vrf[:] = 1500
calls = [call('vrf', 0, 1500), call('vrf', 1, 1500), call('vrf', 2, 1500)]
m.assert_has_calls(calls)
assert m.call_count == 3
def test_set_eiger_default(mocker):
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 2
m = mocker.patch.object(DetectorApi, 'setDac', autospec=True)
# m.return_value = 1560
d = Eiger()
d.dacs.set_default()
calls = [
call('vsvp', 0, 0),
call('vsvp', 1, 0),
call('vtr', 0, 2500),
call('vtr', 1, 2500),
call('vrf', 0, 3300),
call('vrf', 1, 3300),
call('vrs', 0, 1400),
call('vrs', 1, 1400),
call('vsvn', 0, 4000),
call('vsvn', 1, 4000),
call('vtgstv', 0, 2556),
call('vtgstv', 1, 2556),
call('vcmp_ll', 0, 1500),
call('vcmp_ll', 1, 1500),
call('vcmp_lr', 0, 1500),
call('vcmp_lr', 1, 1500),
call('vcall', 0, 4000),
call('vcall', 1, 4000),
call('vcmp_rl', 0, 1500),
call('vcmp_rl', 1, 1500),
call('rxb_rb', 0, 1100),
call('rxb_rb', 1, 1100),
call('rxb_lb', 0, 1100),
call('rxb_lb', 1, 1100),
call('vcmp_rr', 0, 1500),
call('vcmp_rr', 1, 1500),
call('vcp', 0, 200),
call('vcp', 1, 200),
call('vcn', 0, 2000),
call('vcn', 1, 2000),
call('vis', 0, 1550),
call('vis', 1, 1550),
call('iodelay', 0, 660),
call('iodelay', 1, 660)
]
m.assert_has_calls(calls)
assert m.call_count == 17 * 2
def test_set_dac_from_element_in_numpy_array_using_slice(mocker):
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 2
m = mocker.patch.object(DetectorApi, 'setDac', autospec=True)
d = Eiger()
vrf = np.array((1600, 1700, 1800))
d.dacs.vrf[:] = vrf[0]
calls = [
call('vrf', 0, 1600),
call('vrf', 1, 1600),
]
m.assert_has_calls(calls)
assert m.call_count == 2
def test_set_vrf_for_four_mod_different_value_slice(mocker):
m2 = mocker.patch.object(DetectorApi,
'getNumberOfDetectors',
autospec=True)
m2.return_value = 4
m = mocker.patch.object(DetectorApi, 'setDac', autospec=True)
# m.return_value = 1560
d = Eiger()
d.dacs.vrf[:] = [1500, 1600, 1800, 1502]
calls = [
call('vrf', 0, 1500),
call('vrf', 1, 1600),
call('vrf', 2, 1800),
call('vrf', 3, 1502)
]
m.assert_has_calls(calls)
assert m.call_count == 4
def eiger():
from sls_detector import Eiger
return Eiger()
def __init__(self):
self.new_client = Eiger()
self.old_client = DetectorClient()