diff options
Diffstat (limited to 'python/subunit/tests/test_subunit_stats.py')
-rw-r--r-- | python/subunit/tests/test_subunit_stats.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/python/subunit/tests/test_subunit_stats.py b/python/subunit/tests/test_subunit_stats.py index 7c5e42d..9faf24d 100644 --- a/python/subunit/tests/test_subunit_stats.py +++ b/python/subunit/tests/test_subunit_stats.py @@ -18,7 +18,11 @@ import unittest -from testtools.compat import _b, BytesIO, StringIO +from testtools.compat import _b +try: + from testtools.compat import BytesIO, StringIO +except ImportError: + from io import BytesIO, StringIO import subunit |