diff options
Diffstat (limited to 'tests/test_multithreading.py')
| -rw-r--r-- | tests/test_multithreading.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/test_multithreading.py b/tests/test_multithreading.py index b566837..81b0766 100644 --- a/tests/test_multithreading.py +++ b/tests/test_multithreading.py @@ -15,7 +15,12 @@ import sys import time -import mock + +try: + from unittest import mock +except ImportError: + import mock + import testtools import threading import six |