Fix failing unit test

This commit is contained in:
Nick Hall 2015-03-14 20:51:08 +00:00
parent 7147236403
commit 2ecf1a3ddc

View File

@ -58,6 +58,7 @@ class User(user.User):
user.User.__init__(self, callback, error)
self.steps = 0;
self.current_step = 0;
self._input = input
def yes(*args):
return True
@ -135,7 +136,7 @@ class User(user.User):
n = reject_label)
print (text, file = self._fileout) # TODO python3 add flush=True
try:
reply = input()
reply = self._input()
return reply == "" or reply == accept_label
except EOFError:
return False