From eb966bc1605dca3b3715639def60c6d9f845f91e Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Tue, 31 May 2016 21:35:28 -0400 Subject: Add a method to support unittest --- coverage/test_helpers.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'coverage/test_helpers.py') diff --git a/coverage/test_helpers.py b/coverage/test_helpers.py index a76bed35..9649ed5d 100644 --- a/coverage/test_helpers.py +++ b/coverage/test_helpers.py @@ -36,6 +36,10 @@ class Tee(object): for f in self._files: f.flush() + def getvalue(self): + """StringIO file-likes have .getvalue()""" + return self._files[0].getvalue() + if 0: # Use this if you need to use a debugger, though it makes some tests # fail, I'm not sure why... -- cgit v1.2.1