def test_usbtmccomm_query(mock_usbtmc):
comm = USBTMCCommunicator()
comm._filelike.ask = mock.MagicMock(return_value="answer")
eq_(comm._query("mock"), "answer")
comm._filelike.ask.assert_called_with("mock", num=-1, encoding="utf-8")
comm._query("mock", size=10)
comm._filelike.ask.assert_called_with("mock", num=10, encoding="utf-8")
def test_usbtmccomm_read_raw(mock_usbtmc):
comm = USBTMCCommunicator()
comm._filelike.read_raw = mock.MagicMock(return_value=b"abc")
eq_(comm.read_raw(), b"abc")
comm._filelike.read_raw.assert_called_with(num=-1)
assert comm._filelike.read_raw.call_count == 1
comm._filelike.read_raw = mock.MagicMock()
comm.read_raw(10)
comm._filelike.read_raw.assert_called_with(num=10)
def test_usbtmccomm_terminator_setter(mock_usbtmc):
comm = USBTMCCommunicator()
term_char = mock.PropertyMock(return_value="\n")
type(comm._filelike).term_char = term_char
comm.terminator = "*"
assert comm._terminator == "*"
term_char.assert_called_with(42)
comm.terminator = b"*"
assert comm._terminator == "*"
term_char.assert_called_with(42)
def test_usbtmccomm_terminator_setter(mock_usbtmc):
comm = USBTMCCommunicator()
term_char = mock.PropertyMock(return_value="\n")
type(comm._filelike).term_char = term_char
comm.terminator = "*"
eq_(comm._terminator, "*")
term_char.assert_called_with(42)
comm.terminator = b"*" # pylint: disable=redefined-variable-type
eq_(comm._terminator, "*")
term_char.assert_called_with(42)
def test_usbtmccomm_terminator_setter(mock_usbtmc):
comm = USBTMCCommunicator()
term_char = mock.PropertyMock(return_value="\n")
type(comm._filelike).term_char = term_char
comm.terminator = "*"
assert comm._terminator == "*"
term_char.assert_called_with(42)
comm.terminator = b"*"
assert comm._terminator == "*"
term_char.assert_called_with(42)
def test_usbtmccomm_timeout(mock_usbtmc):
comm = USBTMCCommunicator()
timeout = mock.PropertyMock(return_value=1)
type(comm._filelike).timeout = timeout
unit_eq(comm.timeout, 1 * pq.second)
timeout.assert_called_with()
comm.timeout = 10
timeout.assert_called_with(array(10000.0))
comm.timeout = 1000 * pq.millisecond
timeout.assert_called_with(array(1000.0))
def test_usbtmccomm_terminator_getter(mock_usbtmc):
comm = USBTMCCommunicator()
term_char = mock.PropertyMock(return_value="\n")
type(comm._filelike).term_char = term_char
eq_(comm.terminator, "\n")
term_char.assert_called_with()
def open_usbtmc(cls, *args, **kwargs):
"""
Opens an instrument, connecting to a USB-TMC device using the Python
`usbtmc` library.
.. warning:: The operational status of this is unknown. It is suggested
that you connect via the other provided class methods. For Linux,
if you have the ``usbtmc`` kernel module, the
`~instruments.Instrument.open_file` class method will work. On
Windows, using the `~instruments.Instrument.open_visa` class
method along with having the VISA libraries installed will work.
:return: Object representing the connected instrument
"""
usbtmc_comm = USBTMCCommunicator(*args, **kwargs)
return cls(usbtmc_comm)
def test_usbtmccomm_init_missing_module():
_ = USBTMCCommunicator()
def test_usbtmccomm_init(mock_usbtmc):
_ = USBTMCCommunicator("foobar", var1=123)
mock_usbtmc.Instrument.assert_called_with("foobar", var1=123)
def test_usbtmccomm_flush_input(mock_usbtmc):
comm = USBTMCCommunicator()
comm.flush_input()
def test_usbtmccomm_tell(mock_usbtmc):
comm = USBTMCCommunicator()
comm.tell()
def test_usbtmccomm_close(mock_usbtmc):
comm = USBTMCCommunicator()
comm.close()
comm._filelike.close.assert_called_with()
def test_usbtmccomm_sendcmd(mock_usbtmc):
comm = USBTMCCommunicator()
comm.write = mock.MagicMock()
comm._sendcmd("mock")
comm.write.assert_called_with("mock")
def test_usbtmccomm_write_raw(mock_usbtmc):
comm = USBTMCCommunicator()
comm.write_raw(b"mock")
comm._filelike.write_raw.assert_called_with(b"mock")
def open_usbtmc(cls, *args, **kwargs):
# TODO: docstring
usbtmc_comm = USBTMCCommunicator(*args, **kwargs)
return cls(usbtmc_comm)
def test_usbtmccomm_init_missing_module():
with pytest.raises(ImportError):
_ = USBTMCCommunicator()
def test_usbtmccomm_tell(mock_usbtmc):
with pytest.raises(NotImplementedError):
comm = USBTMCCommunicator()
comm.tell()
def test_usbtmccomm_seek(mock_usbtmc):
comm = USBTMCCommunicator()
comm.seek(1)
def test_usbtmccomm_tell(mock_usbtmc):
with pytest.raises(NotImplementedError):
comm = USBTMCCommunicator()
comm.tell()
