diff options
Diffstat (limited to 'Lib/test/test_unittest/support.py')
| -rw-r--r-- | Lib/test/test_unittest/support.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/Lib/test/test_unittest/support.py b/Lib/test/test_unittest/support.py index 5292653..8c97bf5 100644 --- a/Lib/test/test_unittest/support.py +++ b/Lib/test/test_unittest/support.py @@ -136,3 +136,19 @@ class ResultWithNoStartTestRunStopTestRun(object): def wasSuccessful(self): return True + + +class BufferedWriter: + def __init__(self): + self.result = '' + self.buffer = '' + + def write(self, arg): + self.buffer += arg + + def flush(self): + self.result += self.buffer + self.buffer = '' + + def getvalue(self): + return self.result |
