From c1a899ae21323c2f913c640f4688ae18de7aa40a Mon Sep 17 00:00:00 2001 From: Matthieu Huin Date: Tue, 15 Jun 2021 12:11:20 +0200 Subject: 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. --- python/subunit/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'python/subunit/__init__.py') 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 -- cgit v1.2.1