diff options
Diffstat (limited to 'tests/test_bigfile.py')
-rw-r--r-- | tests/test_bigfile.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/test_bigfile.py b/tests/test_bigfile.py index 30156a1..39bd095 100644 --- a/tests/test_bigfile.py +++ b/tests/test_bigfile.py @@ -1,6 +1,9 @@ '''Tests block operations.''' -from StringIO import StringIO +try: + from StringIO import StringIO +except ImportError: + from io import StringIO import unittest import rsa |