diff --git a/test_s3.py b/test_s3.py index 379d381..b7a1f61 100644 --- a/test_s3.py +++ b/test_s3.py @@ -18,7 +18,13 @@ from twisted.python.failure import Failure from twisted.test.proto_helpers import MemoryReactorClock from twisted.trial import unittest -from queue import Queue +import sys +is_py2 = sys.version[0] == '2' +if is_py2: + from Queue import Queue +else: + from queue import Queue + from threading import Event, Thread from mock import Mock