From f96ab609e20b808707bfb507610d019e5f040186 Mon Sep 17 00:00:00 2001
From: Andrew Scheller <github@loowis.durge.org>
Date: Fri, 10 Jun 2016 02:28:44 +0100
Subject: [PATCH] TrafficLights.value also reports 'yellow' instead of 'amber'
 when appropriate

---
 gpiozero/boards.py   | 26 ++++++++++++++++++--------
 tests/test_boards.py | 14 ++++++++++++++
 2 files changed, 32 insertions(+), 8 deletions(-)

diff --git a/gpiozero/boards.py b/gpiozero/boards.py
index 7851988..5d4c3f3 100644
--- a/gpiozero/boards.py
+++ b/gpiozero/boards.py
@@ -12,6 +12,7 @@ except ImportError:
 from time import sleep
 from itertools import repeat, cycle, chain
 from threading import Lock
+from collections import OrderedDict
 
 from .exc import (
     DeviceClosed,
@@ -578,24 +579,33 @@ class TrafficLights(LEDBoard):
             raise OutputDeviceBadValue(
                 'Only one of amber or yellow can be specified'
             )
-        if amber is None:
-            amber = yellow
-        if not all(p is not None for p in [red, amber, green]):
+        devices = OrderedDict((('red', red), ))
+        self._display_yellow = amber is None and yellow is not None
+        if self._display_yellow:
+            devices['yellow'] = yellow
+        else:
+            devices['amber'] = amber
+        devices['green'] = green
+        if not all(p is not None for p in devices.values()):
             raise GPIOPinMissing(
-                'red, amber and green pins must be provided'
+                ', '.join(devices.keys())+' pins must be provided'
             )
         super(TrafficLights, self).__init__(
-            red=red, amber=amber, green=green,
             pwm=pwm, initial_value=initial_value,
-            _order=('red', 'amber', 'green'))
+            _order=devices.keys(),
+            **devices)
 
     def __getattr__(self, name):
-        if name == 'yellow':
+        if name == 'amber' and self._display_yellow:
+            name = 'yellow'
+        elif name == 'yellow' and not self._display_yellow:
             name = 'amber'
         return super(TrafficLights, self).__getattr__(name)
 
     def __setattr__(self, name, value):
-        if name == 'yellow':
+        if name == 'amber' and self._display_yellow:
+            name = 'yellow'
+        elif name == 'yellow' and not self._display_yellow:
             name = 'amber'
         return super(TrafficLights, self).__setattr__(name, value)
 
diff --git a/tests/test_boards.py b/tests/test_boards.py
index 9b5c72c..d1accde 100644
--- a/tests/test_boards.py
+++ b/tests/test_boards.py
@@ -548,11 +548,19 @@ def test_traffic_lights():
     green_pin = MockPin(4)
     with TrafficLights(red_pin, amber_pin, green_pin) as board:
         board.red.on()
+        assert board.red.value
+        assert not board.amber.value
+        assert not board.yellow.value
+        assert not board.green.value
         assert red_pin.state
         assert not amber_pin.state
         assert not green_pin.state
     with TrafficLights(red=red_pin, yellow=amber_pin, green=green_pin) as board:
         board.yellow.on()
+        assert not board.red.value
+        assert board.amber.value
+        assert board.yellow.value
+        assert not board.green.value
         assert not red_pin.state
         assert amber_pin.state
         assert not green_pin.state
@@ -560,6 +568,12 @@ def test_traffic_lights():
 def test_traffic_lights_bad_init():
     with pytest.raises(ValueError):
         TrafficLights()
+    red_pin = MockPin(2)
+    amber_pin = MockPin(3)
+    green_pin = MockPin(4)
+    yellow_pin = MockPin(5)
+    with pytest.raises(ValueError):
+        TrafficLights(red=red_pin, amber=amber_pin, yellow=yellow_pin, green=green_pin)
 
 def test_pi_traffic():
     pins = [MockPin(n) for n in (9, 10, 11)]