From abc703874bcbabc890ffb8d23264a4d2095d72a6 Mon Sep 17 00:00:00 2001
From: Robert Collins <robertc@robertcollins.net>
Date: Mon, 25 Apr 2011 13:29:31 +1200
Subject: [PATCH] Update stats tests.

---
 python/subunit/tests/test_subunit_stats.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/python/subunit/tests/test_subunit_stats.py b/python/subunit/tests/test_subunit_stats.py
index 778bb7f..6fd3301 100644
--- a/python/subunit/tests/test_subunit_stats.py
+++ b/python/subunit/tests/test_subunit_stats.py
@@ -18,7 +18,7 @@
 
 import unittest
 
-from testtools.compat import StringIO
+from testtools.compat import _b, BytesIO, StringIO
 
 import subunit
 
@@ -29,7 +29,7 @@ class TestTestResultStats(unittest.TestCase):
     def setUp(self):
         self.output = StringIO()
         self.result = subunit.TestResultStats(self.output)
-        self.input_stream = StringIO()
+        self.input_stream = BytesIO()
         self.test = subunit.ProtocolTestCase(self.input_stream)
 
     def test_stats_empty(self):
@@ -40,7 +40,7 @@ class TestTestResultStats(unittest.TestCase):
         self.assertEqual(set(), self.result.seen_tags)
 
     def setUpUsedStream(self):
-        self.input_stream.write("""tags: global
+        self.input_stream.write(_b("""tags: global
 test passed
 success passed
 test failed
@@ -52,7 +52,7 @@ test skipped
 skip skipped
 test todo
 xfail todo
-""")
+"""))
         self.input_stream.seek(0)
         self.test.run(self.result)