summaryrefslogtreecommitdiff
path: root/test/coveragetest.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/coveragetest.py')
-rw-r--r--test/coveragetest.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/test/coveragetest.py b/test/coveragetest.py
index 52fde87a..bdee9185 100644
--- a/test/coveragetest.py
+++ b/test/coveragetest.py
@@ -12,11 +12,13 @@ class Tee(object):
def __init__(self, *files):
"""Make a Tee that writes to all the files in `files.`"""
- self.files = files
+ self._files = files
+ if hasattr(files[0], "encoding"):
+ self.encoding = files[0].encoding
def write(self, data):
"""Write `data` to all the files."""
- for f in self.files:
+ for f in self._files:
f.write(data)