def test_gpibusbcomm_read_raw(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm._file.read_raw = mock.MagicMock(return_value=b"abc") eq_(comm.read_raw(3), b"abc") comm._file.read_raw.assert_called_with(3)
def test_gpibusbcomm_sendcmd_empty_string(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm._file.sendcmd = mock.MagicMock() # Refreshed because init makes calls comm._sendcmd("") comm._file.sendcmd.assert_not_called()
def test_gpibusbcomm_eos_old_firmware(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 4 comm._file.sendcmd = mock.MagicMock() comm.eos = "\n" eq_(comm._eos, 10) comm._file.sendcmd.assert_called_with("+eos:10")
def test_gpibusbcomm_timeout(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 unit_eq(comm.timeout, 1000 * pq.millisecond) comm.timeout = 5000 * pq.millisecond comm._file.sendcmd.assert_called_with("++read_tmo_ms 5000.0")
def test_gpibusbcomm_eos_r(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm._file.sendcmd = mock.MagicMock() comm.eos = "\r" assert comm.eos == "\r" assert comm._eos == "\r" comm._file.sendcmd.assert_called_with("++eos 1")
def test_gpibusbcomm_eos_n(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm._file.sendcmd = mock.MagicMock() comm.eos = "\n" eq_(comm.eos, "\n") eq_(comm._eos, "\n") comm._file.sendcmd.assert_called_with("++eos 2")
def test_gpibusbcomm_query_no_question_mark(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm._file.sendcmd = mock.MagicMock() # Refreshed because init makes calls comm._file.read = mock.MagicMock(return_value="answer") comm.sendcmd = mock.MagicMock() eq_(comm._query("mock"), "answer") comm.sendcmd.assert_called_with("mock") comm._file.read.assert_called_with(-1) comm._file.sendcmd.assert_has_calls([mock.call("+read")])
def test_gpibusbcomm_sendcmd(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm._sendcmd("mock") comm._file.sendcmd.assert_has_calls([ mock.call("+a:1"), mock.call("++eoi 1"), mock.call("++read_tmo_ms 1000.0"), mock.call("++eos 2"), mock.call("mock") ])
def test_gpibusbcomm_query(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm._file.read = mock.MagicMock(return_value="answer") comm.sendcmd = mock.MagicMock() eq_(comm._query("mock?"), "answer") comm.sendcmd.assert_called_with("mock?") comm._file.read.assert_called_with(-1) comm._query("mock?", size=10) comm._file.read.assert_called_with(10)
def test_gpibusbcomm_terminator(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 # Default terminator should be eoi assert comm.terminator == "eoi" assert comm._eoi is True comm.terminator = "\n" assert comm.terminator == "\n" assert comm._eoi is False comm.terminator = "eoi" assert comm.terminator == "eoi" assert comm._eoi is True
def test_gpibusbcomm_eoi_old_firmware(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 4 comm._file.sendcmd = mock.MagicMock() comm.eoi = True assert comm.eoi is True assert comm._eoi is True comm._file.sendcmd.assert_called_with("+eoi:1") comm._file.sendcmd = mock.MagicMock() comm.eoi = False assert comm.eoi is False assert comm._eoi is False comm._file.sendcmd.assert_called_with("+eoi:0")
def test_gpibusbcomm_terminator(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 # Default terminator should be eoi eq_(comm.terminator, "eoi") eq_(comm._eoi, True) comm.terminator = "\n" eq_(comm.terminator, "\n") eq_(comm._eoi, False) comm.terminator = "eoi" eq_(comm.terminator, "eoi") eq_(comm._eoi, True)
def test_gpibusbcomm_eoi(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm._file.sendcmd = mock.MagicMock() comm.eoi = True eq_(comm.eoi, True) eq_(comm._eoi, True) comm._file.sendcmd.assert_called_with("++eoi 1") comm._file.sendcmd = mock.MagicMock() comm.eoi = False eq_(comm.eoi, False) eq_(comm._eoi, False) comm._file.sendcmd.assert_called_with("++eoi 0")
def test_gpibusbcomm_eos_invalid(): with pytest.raises(ValueError): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm.eos = "*"
def test_gpibusbcomm_eoi_bad_type(): with pytest.raises(TypeError): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm.eoi = "abc"
def test_gpibusbcomm_eoi_bad_type(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm.eoi = "abc"
def test_gpibusbcomm_write_raw(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm.write_raw(b"mock") comm._file.write_raw.assert_called_with(b"mock")
def test_gpibusbcomm_eos_invalid(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm.eos = "*"
def test_gpibusbcomm_close(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm.close() comm._file.close.assert_called_with()
def test_serialcomm_flush_input(): comm = GPIBCommunicator(mock.MagicMock(), 1) comm._version = 5 comm.flush_input() comm._file.flush_input.assert_called_with()