Tweak Python3 support to help testrepository get 3-ready.

1 files changed, 3 insertions, 3 deletions

diff --git a/python/subunit/__init__.py b/python/subunit/__init__.py
index 42dcf29..dfc5e94 100644
--- a/python/subunit/__init__.py
+++ b/python/subunit/__init__.py
@@ -121,9 +121,9 @@ import re
 import subprocess
 import sys
 import unittest
-if sys.version_info > (3, 0):
+try:
     from io import UnsupportedOperation as _UnsupportedOperation
-else:
+except ImportError:
     _UnsupportedOperation = AttributeError
 
 
@@ -1285,7 +1285,7 @@ def _make_stream_binary(stream):
     """
     try:
         fileno = stream.fileno()
-    except _UnsupportedOperation:
+    except (_UnsupportedOperation, AttributeError):
         pass
     else:
         _make_binary_on_windows(fileno)