def test_serialcomm_close():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm.close()
comm._conn.shutdown.assert_called_with()
comm._conn.close.assert_called_with()
def test_serialcomm_close():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm.close()
comm._conn.shutdown.assert_called_with()
comm._conn.close.assert_called_with()
def test_serialcomm_read_raw_timeout():
with pytest.raises(IOError):
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm._conn.read = mock.MagicMock(side_effect=[b"a", b"b", b""])
_ = comm.read_raw(-1)
def test_loopbackcomm_read_raw_2char_terminator():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm._conn.read = mock.MagicMock(side_effect=[b"a", b"b", b"c", b"\r", b"\n"])
comm._terminator = "\r\n"
eq_(comm.read_raw(), b"abc")
comm._conn.read.assert_has_calls([mock.call(1)] * 5)
assert comm._conn.read.call_count == 5
def test_loopbackcomm_read_raw_2char_terminator():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm._conn.read = mock.MagicMock(
side_effect=[b"a", b"b", b"c", b"\r", b"\n"])
comm._terminator = "\r\n"
assert comm.read_raw() == b"abc"
comm._conn.read.assert_has_calls([mock.call(1)] * 5)
assert comm._conn.read.call_count == 5
def test_serialcomm_address():
# Create our communicator
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
port_name = mock.PropertyMock(return_value="/dev/address")
type(comm._conn).port = port_name
# Check that our address function is working
assert comm.address == "/dev/address"
port_name.assert_called_with()
def test_serialcomm_address():
# Create our communicator
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
port_name = mock.PropertyMock(return_value="/dev/address")
type(comm._conn).port = port_name
# Check that our address function is working
eq_(comm.address, "/dev/address")
port_name.assert_called_with()
def test_serialcomm_query():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm.read = mock.MagicMock(return_value="answer")
comm.sendcmd = mock.MagicMock()
assert comm._query("mock") == "answer"
comm.sendcmd.assert_called_with("mock")
comm.read.assert_called_with(-1)
comm._query("mock", size=10)
comm.read.assert_called_with(10)
def test_serialcomm_read_raw():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm._conn.read = mock.MagicMock(side_effect=[b"a", b"b", b"c", b"\n"])
assert comm.read_raw() == b"abc"
comm._conn.read.assert_has_calls([mock.call(1)] * 4)
assert comm._conn.read.call_count == 4
comm._conn.read = mock.MagicMock()
comm.read_raw(10)
comm._conn.read.assert_called_with(10)
def test_serialcomm_read_raw():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm._conn.read = mock.MagicMock(side_effect=[b"a", b"b", b"c", b"\n"])
eq_(comm.read_raw(), b"abc")
comm._conn.read.assert_has_calls([mock.call(1)]*4)
assert comm._conn.read.call_count == 4
comm._conn.read = mock.MagicMock()
comm.read_raw(10)
comm._conn.read.assert_called_with(10)
def test_serialcomm_query():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm.read = mock.MagicMock(return_value="answer")
comm.sendcmd = mock.MagicMock()
eq_(comm._query("mock"), "answer")
comm.sendcmd.assert_called_with("mock")
comm.read.assert_called_with(-1)
comm._query("mock", size=10)
comm.read.assert_called_with(10)
def test_serialcomm_timeout():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
timeout = mock.PropertyMock(return_value=30)
type(comm._conn).timeout = timeout
unit_eq(comm.timeout, 30 * pq.second)
timeout.assert_called_with()
comm.timeout = 10
timeout.assert_called_with(10)
comm.timeout = 1000 * pq.millisecond
timeout.assert_called_with(1)
def test_serialcomm_timeout():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
timeout = mock.PropertyMock(return_value=30)
type(comm._conn).timeout = timeout
unit_eq(comm.timeout, 30 * u.second)
timeout.assert_called_with()
comm.timeout = 10
timeout.assert_called_with(10)
comm.timeout = 1000 * u.millisecond
timeout.assert_called_with(1)
def test_gpibusbcomm_init():
serial_comm = SerialCommunicator(serial.Serial())
serial_comm._conn = mock.MagicMock()
serial_comm._query = mock.MagicMock(return_value="1")
comm = GPIBCommunicator(serial_comm, 1)
assert isinstance(comm._file, SerialCommunicator)
def test_serialcomm_sendcmd():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm._sendcmd("mock")
comm._conn.write.assert_called_with(b"mock\n")
def test_serialcomm_flush_input():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm.flush_input()
comm._conn.flushInput.assert_called_with()
def test_serialcomm_flush_input():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm.flush_input()
comm._conn.flushInput.assert_called_with()
def test_serialcomm_sendcmd():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm._sendcmd("mock")
comm._conn.write.assert_called_with(b"mock\n")
def test_serialcomm_read_raw_timeout():
comm = SerialCommunicator(serial.Serial())
comm._conn = mock.MagicMock()
comm._conn.read = mock.MagicMock(side_effect=[b"a", b"b", b""])
_ = comm.read_raw(-1)
def test_gpibusbcomm_init():
serial_comm = SerialCommunicator(serial.Serial())
serial_comm._conn = mock.MagicMock()
serial_comm._query = mock.MagicMock(return_value="1")
comm = GPIBCommunicator(serial_comm, 1)
assert isinstance(comm._file, SerialCommunicator)
