Fix #354

Actually, "fix" doesn't really do this justice. This is closer to "nukes
from orbit" ...

tests/test_pins_data.py

@@ -7,13 +7,14 @@ from __future__ import (
 str = type('')
 
 
+import re
 import pytest
 from mock import patch, MagicMock
 
 import gpiozero.devices
 import gpiozero.pins.data
 import gpiozero.pins.native
-from gpiozero.pins.data import pi_info
+from gpiozero.pins.data import pi_info, Style, HeaderInfo, PinInfo
 from gpiozero import PinMultiplePins, PinNoPins, PinUnknownPi
 
 
@@ -88,3 +89,116 @@ def test_pulled_up():
     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_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 'P1:\n' in s
+        assert 'P5:\n' not in s
+
+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['P1']) == '1o'
+    assert '{0:row2}'.format(pi_info('900092').headers['P1']) == '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'])