diff options
| author | Matthieu Huin <mhuin@redhat.com> | 2021-06-15 12:11:20 +0200 |
|---|---|---|
| committer | Matthieu Huin <mhuin@redhat.com> | 2021-06-15 12:11:20 +0200 |
| commit | c1a899ae21323c2f913c640f4688ae18de7aa40a (patch) | |
| tree | 135a6c1b6609f0d67549ce6dcd19d92842cfc802 /python/subunit/tests/test_test_protocol.py | |
| parent | 1dafb884e47f5fc26232672b01c2a9574577e7be (diff) | |
| download | subunit-git-c1a899ae21323c2f913c640f4688ae18de7aa40a.tar.gz | |
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.
Diffstat (limited to 'python/subunit/tests/test_test_protocol.py')
| -rw-r--r-- | python/subunit/tests/test_test_protocol.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/python/subunit/tests/test_test_protocol.py b/python/subunit/tests/test_test_protocol.py index 86aa4a6..faab93e 100644 --- a/python/subunit/tests/test_test_protocol.py +++ b/python/subunit/tests/test_test_protocol.py @@ -21,7 +21,11 @@ import tempfile import unittest from testtools import PlaceHolder, skipIf, TestCase, TestResult -from testtools.compat import _b, _u, BytesIO +from testtools.compat import _b, _u +try: + from testtools.compat import BytesIO, StringIO +except ImportError: + from io import BytesIO, StringIO from testtools.content import Content, TracebackContent, text_content from testtools.content_type import ContentType try: |