Fix imports from testtools.compat

The future release of testtools will drop StringIO and BytesIO.
Try to import them from testtools.compat, or from the io module
if they're not available.

1 files changed, 5 insertions, 1 deletions

diff --git a/python/subunit/__init__.py b/python/subunit/__init__.py
index 17a970a..50b5f23 100644
--- a/python/subunit/__init__.py
+++ b/python/subunit/__init__.py
@@ -129,7 +129,11 @@ except ImportError:
 from extras import safe_hasattr
 from testtools import content, content_type, ExtendedToOriginalDecorator
 from testtools.content import TracebackContent
-from testtools.compat import _b, _u, BytesIO, StringIO
+from testtools.compat import _b, _u, 
+try:
+    from testtools.compat import BytesIO, StringIO
+except ImportError:
+    from io import BytesIO, StringIO
 try:
     from testtools.testresult.real import _StringException
     RemoteException = _StringException