Python version compatibility fixes.

[third_party/subunit] / python / subunit / tests / test_output_filter.py

diff --git a/python/subunit/tests/test_output_filter.py b/python/subunit/tests/test_output_filter.py
index ba966870c5dc5c676d1ba2397426806a0c5fef3a..658174c189ea7c16694cf5e30655b66cf7438d08 100644 (file)
--- a/python/subunit/tests/test_output_filter.py
+++ b/python/subunit/tests/test_output_filter.py
@@ -13,10 +13,11 @@
 #  license you chose for the specific language governing permissions and
 #  limitations under that license.
 #
-import optparse
 import datetime
 from functools import partial
 from io import BytesIO, StringIO
+import optparse
+import sys
 from tempfile import NamedTemporaryFile
 
 from testscenarios import WithScenarios
@@ -127,7 +128,10 @@ class ArgParserTests(TestCase):
     def setUp(self):
         super(ArgParserTests, self).setUp()
         # prevent OptionParser from printing to stderr:
-        self._stderr = BytesIO()
+        if sys.version[0] > '2':
+            self._stderr = StringIO()
+        else:
+            self._stderr = BytesIO()
         self.patch(optparse.sys, 'stderr', self._stderr)
 
     def test_can_parse_attach_file_without_test_id(self):