def test_property_creation_with_blanks(cmd_input, cmd_kwds, expected):
subsys = scpi.get_blank_scpi_subsystem('Properties')
s = subsys(Mock())
d = set(s.__dir__())
scpi.add_scpi_cmd(subsys, *cmd_input, **cmd_kwds)
assert expected == set(s.__dir__()).difference(d)
def test_qty_converter_unit():
subsys = scpi.get_blank_scpi_subsystem('QtyClass')
scpi.add_scpi_cmd(subsys,
'frequency',
'cmd:frequency',
scpi.scpi_qty('Hz'),
query_keywords=['min', 'max'],
write_keywords=['min', 'max', 'default'])
p = Mock(**{'device.query.return_value': '500'})
s = subsys(p)
s.frequency = 100 * ureg.mHz
assert 'cmd:frequency 0.1' == p.device.write.call_args.args[0]
p.reset_mock()
with pytest.raises(pint.errors.DimensionalityError):
s.frequency = 234 * ureg.m
p.device.write.assert_not_called()
with pytest.raises(ValueError):
s.frequency = 234
p.device.write.assert_not_called()
assert 500 * ureg.Hz == s.frequency
assert 'cmd:frequency?' == p.device.query.call_args.args[0]
def test_boolean_converter_with_keywords():
subsys = scpi.get_blank_scpi_subsystem('BooleanSubsystem')
scpi.add_scpi_cmd(subsys,
'enabled',
'cmd:subcmd',
scpi.scpi_bool(),
query_keywords=['qmin', 'qmax'],
write_keywords=['wkw'])
p = Mock(**{'device.query.return_value': '1'})
s = subsys(p)
s.enabled = True
assert 'cmd:subcmd 1' == p.device.write.call_args.args[0]
s.enabled = False
assert 'cmd:subcmd 0' == p.device.write.call_args.args[0]
s.enabled_to_wkw()
assert 'cmd:subcmd wkw' == p.device.write.call_args.args[0]
assert True is s.enabled
assert 'cmd:subcmd?' == p.device.query.call_args.args[0]
assert True is s.enabled_qmin
assert 'cmd:subcmd? qmin' == p.device.query.call_args.args[0]
assert True is s.enabled_qmax
assert 'cmd:subcmd? qmax' == p.device.query.call_args.args[0]
def test_argless_command():
subsys = scpi.get_blank_scpi_subsystem('ArglessCommand')
scpi.add_scpi_cmd(subsys, 'clear', 'display:clear', None)
p = Mock()
s = subsys(p)
s.clear()
assert 'display:clear' == p.device.write.call_args.args[0]
def test_error_converter():
subsys = scpi.get_blank_scpi_subsystem('ErrorSubsystem')
scpi.add_scpi_cmd(subsys, 'error', 'system:error', scpi.scpi_error())
s = subsys(Mock(**{'device.query.return_value': '+0,"No Error"'}))
assert (0, 'No Error') == s.error
with pytest.raises(AttributeError):
s.error = 'bad' # Can not write errors back to device.
def test_calling_with_extra_fields():
subsys = scpi.get_blank_scpi_subsystem('Test')
p = Mock()
s = subsys(p)
scpi.add_scpi_cmd(subsys, 'noarg', 'cmd:ch{ch}', None)
with pytest.raises(KeyError):
s.noarg()
s.keys['ch'] = 1
s.noarg()
assert 'cmd:ch1' == p.device.write.call_args.args[0]
def test_boolean_converter():
subsys = scpi.get_blank_scpi_subsystem('BooleanSubsystem')
scpi.add_scpi_cmd(subsys, 'enabled', 'cmd:subcmd', scpi.scpi_bool())
p = Mock(**{'device.query.return_value': '1'})
s = subsys(p)
s.enabled = True
assert 'cmd:subcmd 1' == p.device.write.call_args.args[0]
s.enabled = False
assert 'cmd:subcmd 0' == p.device.write.call_args.args[0]
s.enabled = 1
assert 'cmd:subcmd 1' == p.device.write.call_args.args[0]
s.enabled = 0
assert 'cmd:subcmd 0' == p.device.write.call_args.args[0]
assert True is s.enabled
assert 'cmd:subcmd?' == p.device.query.call_args.args[0]
def test_choices_converter():
subsys = scpi.get_blank_scpi_subsystem('ChoicesClass')
scpi.add_scpi_cmd(subsys, 'shape', 'cmd:shape',
scpi.scpi_choice({
'sinusoid': 'SIN',
'square': 'SQU'
}))
p = Mock(**{'device.query.return_value': 'SQU'})
s = subsys(p)
s.shape = 'sinusoid'
assert 'cmd:shape SIN' == p.device.write.call_args.args[0]
assert s.shape == 'square'
assert 'cmd:shape?' == p.device.query.call_args.args[0]
with pytest.raises(KeyError):
s.shape = 'triangle'