tristate: fix __repr__

and add doctest

1 files changed, 15 insertions, 5 deletions

diff --git a/gbp/tristate.py b/gbp/tristate.py
index 95391b90..67b1052d 100644
--- a/gbp/tristate.py
+++ b/gbp/tristate.py
@@ -43,12 +43,22 @@ class Tristate(object):
             raise TypeError
 
     def __repr__(self):
-        if self._state == ON:
-            return "on"
-        elif self._state == AUTO:
-            return "auto"
+        """
+        >>> Tristate('on').__repr__()
+        'on'
+        >>> Tristate(True).__repr__()
+        'on'
+        >>> Tristate(False).__repr__()
+        'off'
+        >>> Tristate('auto').__repr__()
+        'auto'
+        """
+        if self._state == self.ON:
+            return 'on'
+        elif self._state == self.AUTO:
+            return 'auto'
         else:
-            return "off"
+            return 'off'
 
     @classmethod
     def is_valid_state(self, stat):