diff options
-rw-r--r-- | gbp/tristate.py | 20 |
1 files changed, 15 insertions, 5 deletions
diff --git a/gbp/tristate.py b/gbp/tristate.py index 95391b9..67b1052 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): |