diff --git a/test_requests.py b/test_requests.py index 963abbc1..a45f5cb2 100644 --- a/test_requests.py +++ b/test_requests.py @@ -1,12 +1,16 @@ #!/usr/bin/env python # -*- coding: utf-8 -*- +import sys import unittest -try: - from cStringIO import StringIO -except ImportError: - from StringIO import StringIO +if sys.version_info >= (3,0): + from io import StringIO +else: + try: + from cStringIO import StringIO + except ImportError: + from StringIO import StringIO import requests