diff options
Diffstat (limited to 'tests/test_thread_safety.py')
-rw-r--r-- | tests/test_thread_safety.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tests/test_thread_safety.py b/tests/test_thread_safety.py index 83a9742..3b90e67 100644 --- a/tests/test_thread_safety.py +++ b/tests/test_thread_safety.py @@ -4,10 +4,12 @@ import os, sys sys.path.insert(0, os.path.abspath('..')) -from threading import Thread -import unittest +try: + import unittest2 as unittest +except ImportError: + import unittest -import requests +from threading import Thread from requests_cache import CachedSession CACHE_NAME = 'requests_cache_test' |