Make all unit tests pass on py3.3/3.4

1 files changed, 9 insertions, 2 deletions

diff --git a/kafka/consumer.py b/kafka/consumer.py
index 0935dd2..7636a83 100644
--- a/kafka/consumer.py
+++ b/kafka/consumer.py
@@ -1,12 +1,19 @@
 from __future__ import absolute_import
 
-from itertools import izip_longest, repeat
+try:
+    from itertools import zip_longest as izip_longest, repeat  # pylint: disable-msg=E0611
+except ImportError:  # python 2
+    from itertools import izip_longest as izip_longest, repeat
 import logging
 import time
 import numbers
 from threading import Lock
 from multiprocessing import Process, Queue as MPQueue, Event, Value
-from Queue import Empty, Queue
+
+try:
+    from Queue import Empty, Queue
+except ImportError:  # python 2
+    from queue import Empty, Queue
 
 import kafka
 from kafka.common import (