From 3a381ec683bda6b8366252967d72217befffbd61 Mon Sep 17 00:00:00 2001 From: Guido Günther Date: Sun, 22 Jan 2012 16:57:02 +0100 Subject: tristate: fix __repr__ and add doctest --- gbp/tristate.py | 20 +++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) (limited to 'gbp/tristate.py') 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): -- cgit v1.2.3