3333#
3434
3535from __future__ import division , unicode_literals , print_function , absolute_import
36+ import collections
37+ import struct
3638
3739from pyvisa import visa
3840from pyvisa .visa import InvalidBinaryFormat
3941import pytest
4042from mock import Mock
41- import collections
42- import struct
4343
4444def packfloats1 (fmt , floats ):
4545 data = b'' .join (struct .pack (fmt , num ) for num in floats )
@@ -57,8 +57,8 @@ def pytest_funcarg__instrument(self, request):
5757 'interface_type' ,
5858 visa .VI_INTF_GPIB )
5959 my_attr = MyAttribute ()
60- monkeypatch .setattr (visa .vpp43 , "get_attribute" , my_attr .get_attribute )
61- monkeypatch .setattr (visa .vpp43 , "set_attribute" , my_attr .set_attribute )
60+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , "get_attribute" , my_attr .get_attribute )
61+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , "set_attribute" , my_attr .set_attribute )
6262 return visa .Instrument (1 )
6363
6464 def test_repr (self , instrument ):
@@ -74,7 +74,7 @@ def test_repr(self, instrument):
7474 (b"hi there\r \n " , b"hi there\r \n " )])
7575 def test_write (self , monkeypatch , instrument , message , expected ):
7676 my_write = Mock ()
77- monkeypatch .setattr (visa .vpp43 , 'write' , my_write )
77+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'write' , my_write )
7878 instrument .write (message )
7979 print (repr (instrument .term_chars ))
8080 my_write .assert_called_with (0 , expected )
@@ -87,7 +87,7 @@ def test_write(self, monkeypatch, instrument, message, expected):
8787 def test_write_termchars_set (self , monkeypatch , instrument ,
8888 message , expected ):
8989 my_write = Mock ()
90- monkeypatch .setattr (visa .vpp43 , 'write' , my_write )
90+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'write' , my_write )
9191 instrument .term_chars = b'\n '
9292 instrument .write (message )
9393 my_write .assert_called_with (0 , expected )
@@ -100,7 +100,7 @@ def test_write_termchars_set(self, monkeypatch, instrument,
100100 def test_write_delay_set (self , monkeypatch , instrument , message , expected ):
101101 my_write = Mock ()
102102 my_sleep = Mock ()
103- monkeypatch .setattr (visa .vpp43 , 'write' , my_write )
103+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'write' , my_write )
104104 monkeypatch .setattr (visa .time , 'sleep' , my_sleep )
105105 instrument .delay = 1
106106 instrument .write (message )
@@ -140,16 +140,16 @@ def test_send_end(self, instrument, value):
140140 def test_read_raw (self , monkeypatch , instrument ):
141141 my_read = Mock (return_value = b'some bytes\r \n ' )
142142 my_get_status = Mock (side_effect = [visa .VI_SUCCESS_MAX_CNT , 0 ])
143- monkeypatch .setattr (visa .vpp43 , 'read' , my_read )
144- monkeypatch .setattr (visa .vpp43 , 'get_status' , my_get_status )
143+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'read' , my_read )
144+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'get_status' , my_get_status )
145145 result = instrument .read_raw ()
146146 assert result == b'some bytes\r \n some bytes\r \n '
147147
148148 def test_read (self , monkeypatch , instrument ):
149149 my_read = Mock (return_value = b'some bytes\r \n ' )
150150 my_get_status = Mock (side_effect = [visa .VI_SUCCESS_MAX_CNT , 0 ])
151- monkeypatch .setattr (visa .vpp43 , 'read' , my_read )
152- monkeypatch .setattr (visa .vpp43 , 'get_status' , my_get_status )
151+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'read' , my_read )
152+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'get_status' , my_get_status )
153153 result = instrument .read ()
154154 assert result == b'some bytes\r \n some bytes'
155155
@@ -204,8 +204,8 @@ def test_read_values(self, monkeypatch, instrument,
204204 format , input , expected ):
205205 my_read = Mock (return_value = input )
206206 my_get_status = Mock (return_value = 0 )
207- monkeypatch .setattr (visa .vpp43 , 'read' , my_read )
208- monkeypatch .setattr (visa .vpp43 , 'get_status' , my_get_status )
207+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'read' , my_read )
208+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'get_status' , my_get_status )
209209 result = instrument .read_values (format )
210210 assert result == expected
211211
@@ -220,8 +220,8 @@ def test_read_values_errors(self, monkeypatch, instrument,
220220 format , input , expected_exception ):
221221 my_read = Mock (return_value = input )
222222 my_get_status = Mock (return_value = 0 )
223- monkeypatch .setattr (visa .vpp43 , 'read' , my_read )
224- monkeypatch .setattr (visa .vpp43 , 'get_status' , my_get_status )
223+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'read' , my_read )
224+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , 'get_status' , my_get_status )
225225 with pytest .raises (expected_exception ):
226226 instrument .read_values (format )
227227
@@ -237,8 +237,8 @@ def pytest_funcarg__instrument(self, request):
237237 visa .VI_INTF_GPIB )
238238 monkeypatch .setattr (visa .GpibInstrument , 'stb' , 0x40 )
239239 my_attr = MyAttribute ()
240- monkeypatch .setattr (visa .vpp43 , "get_attribute" , my_attr .get_attribute )
241- monkeypatch .setattr (visa .vpp43 , "set_attribute" , my_attr .set_attribute )
240+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , "get_attribute" , my_attr .get_attribute )
241+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , "set_attribute" , my_attr .set_attribute )
242242 return visa .GpibInstrument (1 )
243243
244244 @pytest .mark .parametrize ('timeout' , [None , 0 , 25 , 4294967 ])
@@ -266,8 +266,8 @@ def pytest_funcarg__instrument(self, request):
266266 'interface_type' ,
267267 visa .VI_INTF_ASRL )
268268 my_attr = MyAttribute ()
269- monkeypatch .setattr (visa .vpp43 , "get_attribute" , my_attr .get_attribute )
270- monkeypatch .setattr (visa .vpp43 , "set_attribute" , my_attr .set_attribute )
269+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , "get_attribute" , my_attr .get_attribute )
270+ monkeypatch .setattr (pyvisa . wrapper .vpp43 , "set_attribute" , my_attr .set_attribute )
271271 return visa .SerialInstrument (1 )
272272
273273 def test_term_chars_default (self , instrument ):
0 commit comments