Skip to content

Commit a9edf52

Browse files
committed
Moved vpp43 to wrapper subpackage.
See LabPy#13
1 parent ff8cfad commit a9edf52

File tree

10 files changed

+84
-86
lines changed

10 files changed

+84
-86
lines changed

pyvisa/__init__.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,15 @@
1313
"""
1414

1515
from __future__ import division, unicode_literals, print_function, absolute_import
16-
1716
import os
1817
import sys
1918
import logging
20-
import pkg_resources
2119
import subprocess
22-
import ConfigParser
2320

24-
from . import vpp43
21+
import pkg_resources
22+
23+
import ConfigParser
24+
from pyvisa.wrapper import functions
2525

2626
logger = logging.getLogger('pyvisa')
2727
logger.addHandler(logging.NullHandler)
@@ -46,4 +46,4 @@
4646
except ConfigParser.Error:
4747
pass
4848
else:
49-
vpp43.visa_library.load_library(_visa_library_path)
49+
functions.visa_library.load_library(_visa_library_path)

pyvisa/testsuite/test_visa.py

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -33,13 +33,13 @@
3333
#
3434

3535
from __future__ import division, unicode_literals, print_function, absolute_import
36+
import collections
37+
import struct
3638

3739
from pyvisa import visa
3840
from pyvisa.visa import InvalidBinaryFormat
3941
import pytest
4042
from mock import Mock
41-
import collections
42-
import struct
4343

4444
def 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\nsome 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\nsome 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

Comments
 (0)