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
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
|
# vim: set fileencoding=utf-8:
#
# GPIO Zero: a library for controlling the Raspberry Pi's GPIO pins
#
# Copyright (c) 2016-2021 Dave Jones <dave@waveform.org.uk>
# Copyright (c) 2019 Ben Nuttall <ben@bennuttall.com>
# Copyright (c) 2018 Martchus <martchus@gmx.net>
#
# SPDX-License-Identifier: BSD-3-Clause
from __future__ import (
unicode_literals,
absolute_import,
print_function,
division,
)
str = type('')
import io
import re
import errno
import pytest
from mock import patch, MagicMock
import gpiozero.pins.data
import gpiozero.pins.local
from gpiozero.pins.local import LocalPiFactory
from gpiozero.pins.data import Style, HeaderInfo, PinInfo
from gpiozero import *
def test_pi_revision():
with patch('gpiozero.devices.Device.pin_factory', LocalPiFactory()):
# Can't use MockPin for this as we want something that'll actually try
# and read /proc/device-tree/system/linux,revision and /proc/cpuinfo
# (MockPin simply parrots the 3B's data); LocalPiFactory is used as we
# can definitely instantiate it (strictly speaking it's abstract but
# we're only interested in the pi_info stuff)
with patch('io.open') as m:
m.return_value.__enter__.side_effect = [
# Pretend /proc/device-tree/system/linux,revision doesn't
# exist, and that /proc/cpuinfo contains the Revision: 0002
# after some filler
IOError(errno.ENOENT, 'File not found'),
['lots of irrelevant', 'lines', 'followed by', 'Revision: 0002', 'Serial: xxxxxxxxxxx']
]
assert pi_info().revision == '0002'
# LocalPiFactory caches the revision (because realistically it
# isn't going to change at runtime); we need to wipe it here though
Device.pin_factory._info = None
m.return_value.__enter__.side_effect = [
IOError(errno.ENOENT, 'File not found'),
['Revision: a21042']
]
assert pi_info().revision == 'a21042'
# Check over-volting result (some argument over whether this is 7
# or 8 character result; make sure both work)
Device.pin_factory._info = None
m.return_value.__enter__.side_effect = [
IOError(errno.ENOENT, 'File not found'),
['Revision: 1000003']
]
assert pi_info().revision == '0003'
Device.pin_factory._info = None
m.return_value.__enter__.side_effect = [
IOError(errno.ENOENT, 'File not found'),
['Revision: 100003']
]
assert pi_info().revision == '0003'
# Check we complain loudly if we can't access linux,revision
Device.pin_factory._info = None
m.return_value.__enter__.side_effect = [
# Pretend /proc/device-tree/system/linux,revision doesn't
# exist, and that /proc/cpuinfo contains the Revision: 0002
# after some filler
IOError(errno.EACCES, 'Permission denied'),
['Revision: 100003']
]
with pytest.raises(IOError):
pi_info()
# Check that parsing /proc/device-tree/system/linux,revision also
# works properly
Device.pin_factory._info = None
m.return_value.__enter__.side_effect = None
m.return_value.__enter__.return_value = io.BytesIO(b'\x00\xa2\x20\xd3')
assert pi_info().revision == 'a220d3'
# Check that if everything's a bust we raise PinUnknownPi
with pytest.raises(PinUnknownPi):
Device.pin_factory._info = None
m.return_value.__enter__.return_value = None
m.return_value.__enter__.side_effect = [
IOError(errno.ENOENT, 'File not found'),
['nothing', 'relevant']
]
pi_info()
with pytest.raises(PinUnknownPi):
pi_info('0fff')
def test_pi_info():
r = pi_info('900011')
assert r.model == 'B'
assert r.pcb_revision == '1.0'
assert r.memory == 512
assert r.manufacturer == 'Sony'
assert r.storage == 'SD'
assert r.usb == 2
assert r.ethernet == 1
assert not r.wifi
assert not r.bluetooth
assert r.csi == 1
assert r.dsi == 1
r = pi_info('9000f1')
assert r.model == '???'
assert r.pcb_revision == '1.1'
assert r.memory == 512
assert r.manufacturer == 'Sony'
assert r.storage == 'MicroSD'
assert r.usb == 4
assert r.ethernet == 1
assert not r.wifi
assert not r.bluetooth
assert r.csi == 1
assert r.dsi == 1
assert repr(r).startswith('PiBoardInfo(revision=')
assert 'headers=...' in repr(r)
def test_pi_info_not_a_pi():
class NotAPiFactory(LocalPiFactory):
def _get_pi_info(self):
return None
with patch('gpiozero.devices.Device.pin_factory', NotAPiFactory()):
with pytest.raises(PinUnknownPi):
pi_info()
def test_pi_info_other_types():
assert pi_info(b'9000f1') == pi_info(0x9000f1)
def test_physical_pins():
# Assert physical pins for some well-known Pi's; a21041 is a Pi2B
assert pi_info('a21041').physical_pins('3V3') == {('J8', 1), ('J8', 17)}
assert pi_info('a21041').physical_pins('GPIO2') == {('J8', 3)}
assert pi_info('a21041').physical_pins('GPIO47') == set()
def test_physical_pin():
with pytest.raises(PinMultiplePins):
assert pi_info('a21041').physical_pin('GND')
assert pi_info('a21041').physical_pin('GPIO3') == ('J8', 5)
with pytest.raises(PinNoPins):
assert pi_info('a21041').physical_pin('GPIO47')
def test_pulled_up():
assert pi_info('a21041').pulled_up('GPIO2')
assert not pi_info('a21041').pulled_up('GPIO4')
assert not pi_info('a21041').pulled_up('GPIO47')
def test_pprint_content():
with patch('sys.stdout') as stdout:
stdout.output = []
stdout.write = lambda buf: stdout.output.append(buf)
pi_info('900092').pprint(color=False)
s = ''.join(stdout.output)
assert ('o' * 20 + ' ') in s # first header row
assert ('1' + 'o' * 19 + ' ') in s # second header row
assert 'PiZero' in s
assert 'V1.2' in s # PCB revision
assert '900092' in s # Pi revision
assert 'BCM2835' in s # SOC name
stdout.output = []
pi_info('0002').pprint(color=False)
s = ''.join(stdout.output)
assert ('o' * 13 + ' ') in s # first header row
assert ('1' + 'o' * 12 + ' ') in s # second header row
assert 'Pi Model' in s
assert 'B V1.0' in s # PCB revision
assert '0002' in s # Pi revision
assert 'BCM2835' in s # SOC name
stdout.output = []
pi_info('0014').headers['SODIMM'].pprint(color=False)
assert len(''.join(stdout.output).splitlines()) == 100
def test_format_content():
with patch('sys.stdout') as stdout:
stdout.output = []
stdout.write = lambda buf: stdout.output.append(buf)
pi_info('900092').pprint(color=False)
s = ''.join(stdout.output)
assert '{0:mono}\n'.format(pi_info('900092')) == s
stdout.output = []
pi_info('900092').pprint(color=True)
s = ''.join(stdout.output)
assert '{0:color full}\n'.format(pi_info('900092')) == s
with pytest.raises(ValueError):
'{0:color foo}'.format(pi_info('900092'))
def test_pprint_headers():
assert len(pi_info('0002').headers) == 1
assert len(pi_info('000e').headers) == 2
assert len(pi_info('900092').headers) == 1
with patch('sys.stdout') as stdout:
stdout.output = []
stdout.write = lambda buf: stdout.output.append(buf)
pi_info('0002').pprint()
s = ''.join(stdout.output)
assert 'P1:\n' in s
assert 'P5:\n' not in s
stdout.output = []
pi_info('000e').pprint()
s = ''.join(stdout.output)
assert 'P1:\n' in s
assert 'P5:\n' in s
stdout.output = []
pi_info('900092').pprint()
s = ''.join(stdout.output)
assert 'J8:\n' in s
assert 'P1:\n' not in s
assert 'P5:\n' not in s
def test_format_headers():
with patch('sys.stdout') as stdout:
stdout.output = []
stdout.write = lambda buf: stdout.output.append(buf)
info = pi_info('c03131')
info.headers['J8'].pprint(color=False)
s = ''.join(stdout.output)
assert '{0.headers[J8]:mono}\n'.format(info) == s
stdout.output = []
info.headers['J8'].pprint(color=True)
s = ''.join(stdout.output)
assert '{0.headers[J8]:color}\n'.format(info) == s
with pytest.raises(ValueError):
'{0.headers[J8]:mono foo}'.format(info)
def test_pprint_color():
with patch('sys.stdout') as stdout:
stdout.output = []
stdout.write = lambda buf: stdout.output.append(buf)
pi_info('900092').pprint(color=False)
s = ''.join(stdout.output)
assert '\x1b[0m' not in s # make sure ANSI reset code isn't in there
stdout.output = []
pi_info('900092').pprint(color=True)
s = ''.join(stdout.output)
assert '\x1b[0m' in s # check the ANSI reset code *is* in there (can't guarantee much else!)
stdout.output = []
stdout.fileno.side_effect = IOError('not a real file')
pi_info('900092').pprint()
s = ''.join(stdout.output)
assert '\x1b[0m' not in s # default should output mono
with patch('os.isatty') as isatty:
isatty.return_value = True
stdout.fileno.side_effect = None
stdout.output = []
pi_info('900092').pprint()
s = ''.join(stdout.output)
assert '\x1b[0m' in s # default should now output color
def test_pprint_styles():
with pytest.raises(ValueError):
Style.from_style_content('mono color full')
with pytest.raises(ValueError):
Style.from_style_content('full specs')
with patch('sys.stdout') as stdout:
s = '{0:full}'.format(pi_info('900092'))
assert '\x1b[0m' not in s # ensure default is mono when stdout is not a tty
with pytest.raises(ValueError):
'{0:foo on bar}'.format(Style())
def test_pprint_missing_pin():
header = HeaderInfo('FOO', 4, 2, {
1: PinInfo(1, '5V', False, 1, 1),
2: PinInfo(2, 'GND', False, 1, 2),
# Pin 3 is deliberately missing
4: PinInfo(4, 'GPIO1', False, 2, 2),
5: PinInfo(5, 'GPIO2', False, 3, 1),
6: PinInfo(6, 'GPIO3', False, 3, 2),
7: PinInfo(7, '3V3', False, 4, 1),
8: PinInfo(8, 'GND', False, 4, 2),
})
with patch('sys.stdout') as stdout:
stdout.output = []
stdout.write = lambda buf: stdout.output.append(buf)
s = ''.join(stdout.output)
header.pprint()
for i in range(1, 9):
if i == 3:
assert '(3)' not in s
else:
assert ('(%d)' % i)
def test_pprint_rows_cols():
assert '{0:row1}'.format(pi_info('900092').headers['J8']) == '1o'
assert '{0:row2}'.format(pi_info('900092').headers['J8']) == 'oo'
assert '{0:col1}'.format(pi_info('0002').headers['P1']) == '1oooooooooooo'
assert '{0:col2}'.format(pi_info('0002').headers['P1']) == 'ooooooooooooo'
with pytest.raises(ValueError):
'{0:row16}'.format(pi_info('0002').headers['P1'])
with pytest.raises(ValueError):
'{0:col3}'.format(pi_info('0002').headers['P1'])
|
