forked from p3trus/slave
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathtest_b2900.py
163 lines (148 loc) · 6.58 KB
/
test_b2900.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
# -*- coding: utf-8 -*-
# -*- coding: utf-8 -*-
import re
import unittest
from random import randint
from itertools import product
import visa
from slave.agilent import B2900
from slave.transport import Visa
class TestB2902A(unittest.TestCase):
@classmethod
def setUpClass(cls):
transport = Visa(search_b2900())
cls.b2900 = B2900(transport)
cls.b2900r = B2900(transport)
def tearDown(self):
self.b2900.reset()
def test_setup_triggering(self):
for s, cn, t, d, ch in product(
('automatic internal', 'bus', 'timer'),
(1, 100000, randint(1, 100000)),
(2e-5, 1e5, randint(2, 1e10) / 1e5),
(0, 100, randint(0, 10000) / 100),
(1, 2)
):
with self.subTest(source=s, count=cn, timer=t, delay=d, channel=ch):
self.b2900.setup.triggering(s, cn, t, d, ch)
idx = ch - 1
self.assertEqual(self.b2900r.triggerings[idx].transient.source, s)
self.assertEqual(self.b2900r.triggerings[idx].acquire.source, s)
self.assertEqual(self.b2900r.triggerings[idx].transient.count, cn)
self.assertEqual(self.b2900r.triggerings[idx].acquire.count, cn)
self.assertAlmostEqual(self.b2900r.triggerings[idx].transient.timer, t, delta=0.01)
self.assertAlmostEqual(self.b2900r.triggerings[idx].acquire.delay, d, delta=0.01)
def test_setup_fixed_source(self):
self.b2900.reset()
for f, v, c in product(
('voltage', 'current'),
(0, 0.01, randint(0, 100) / 10000, None),
(1, 2)
):
with self.subTest(function=f, value=v, channel=c):
for trig in self.b2900.triggerings:
trig.count = 100
self.b2900.setup.fixed_source(f, v, c)
idx = c - 1
self.assertEqual(self.b2900r.triggerings[idx].transient.count, 1)
self.assertEqual(self.b2900r.triggerings[idx].acquire.count, 1)
self.assertEqual(self.b2900r.sources[idx].function_mode, f)
if v is not None:
if f == 'voltage':
self.assertAlmostEqual(self.b2900r.sources[idx].voltage.level, v, delta=0.001)
self.assertAlmostEqual(self.b2900r.sources[idx].voltage.level_triggered, v, delta=0.001)
else:
self.assertAlmostEqual(self.b2900r.sources[idx].current.level, v, delta=0.001)
self.assertAlmostEqual(self.b2900r.sources[idx].current.level_triggered, v, delta=0.001)
def test_sweep_source(self):
self.b2900.reset()
for f, start, stop, points, ch in product(
('voltage', 'current'),
(0, 0.1),
(0.15, 0.05),
(1, 101, randint(1, 101)),
(1, 2)
):
with self.subTest(function=f, start=start, stop=stop, points=points, channel=ch):
idx = ch - 1
self.b2900.setup.sweep_source(f, start, stop, points, ch)
self.assertEqual(self.b2900r.sources[idx].function_mode, f)
sub = self.b2900r.sources[idx].voltage if f == 'voltage' else self.b2900r.sources[idx].current
self.assertAlmostEqual(sub.sweep_start, start, delta=1e-4)
self.assertAlmostEqual(sub.sweep_stop, stop, delta=1e-4)
self.assertEqual(sub.points, points)
self.assertEqual(self.b2900r.triggerings[idx].transient.count, points)
self.assertEqual(self.b2900r.triggerings[idx].acquire.count, points)
self.b2900.initiate(channels=(ch,))
self.b2900.wait_to_continue()
data = self.b2900.fetch_array((ch,))
if isinstance(data, float):
length = 1
else:
elms = self.b2900r.format.sense_elements
if isinstance(elms, str):
length = len(data)
else:
length = len(data) / len(elms)
self.assertEqual(length, points)
def test_setup_sense(self):
for f, ar, r, nplc, compl, ch in product(
('voltage', 'current'),
(True, False, None),
(0.1, 1, None),
(0.01, 10, None),
(0, 0.01, None),
(1, 2)
):
with self.subTest(function=f, auto_range=ar, range_=r, nplc=nplc, compliance=compl, channel=ch):
if ar and r:
continue
self.b2900.setup.sense(f, ar, r, nplc, compl, ch)
idx = ch - 1
sub = self.b2900r.senses[idx].voltage if f == 'voltage' else self.b2900r.senses[idx].current
if ar is not None:
self.assertEqual(sub.auto_range, ar)
if r is not None:
self.assertGreaterEqual(sub.range, r)
if nplc is not None:
self.assertAlmostEqual(sub.nplc, nplc, delta=0.01)
if compl is not None:
self.assertAlmostEqual(sub.compliance, compl, delta=0.01)
def test_setup_resistance(self):
for e, m, c in product(
(True, False, None),
('auto', 'manual', None),
(1, 2)
):
with self.subTest(enable=e, mode=m, channel=c):
self.b2900.setup.resistance(e, m, c)
idx = c - 1
if e is None:
pass
elif e:
self.assertIn('resistance', self.b2900r.senses[idx].functions)
else:
self.assertNotIn('resistance', self.b2900r.senses[idx].functions)
if m is not None:
self.assertEqual(self.b2900r.senses[idx].resistance.mode, m)
def search_b2900():
rm = visa.ResourceManager()
for res in rm.list_resources():
try:
i = rm.open_resource(res)
except visa.VisaIOError:
continue
else:
try:
i.clear()
identity = i.query('*IDN?').split(',')
except visa.VisaIOError:
continue
if re.search(r'B29[0-9][0-9]', identity[1]):
break
else:
raise RuntimeError('Could not find B290X Source-measure unit.')
return res
if __name__ == '__main__':
if not input('Hit ENTER to start tests. Input something to abort: '):
unittest.main()