diff options
| -rw-r--r-- | kafka/client.py | 3 | ||||
| -rw-r--r-- | test/test_unit.py | 4 |
2 files changed, 4 insertions, 3 deletions
diff --git a/kafka/client.py b/kafka/client.py index a31ea66..965cbc5 100644 --- a/kafka/client.py +++ b/kafka/client.py @@ -1,9 +1,10 @@ from collections import defaultdict from functools import partial +from itertools import count import logging import time -from kafka.common import count, ErrorMapping, TopicAndPartition +from kafka.common import ErrorMapping, TopicAndPartition from kafka.conn import KafkaConnection from kafka.protocol import KafkaProtocol diff --git a/test/test_unit.py b/test/test_unit.py index c796c94..3f3af66 100644 --- a/test/test_unit.py +++ b/test/test_unit.py @@ -3,7 +3,8 @@ import random import struct import unittest -from kafka.client import KafkaClient, ProduceRequest, FetchRequest +from kafka.client import KafkaClient +from kafka.common import ProduceRequest, FetchRequest from kafka.codec import ( has_gzip, has_snappy, gzip_encode, gzip_decode, @@ -59,7 +60,6 @@ class TestMisc(unittest.TestCase): def test_length_prefix(self): for i in xrange(ITERATIONS): s1 = random_string() - s2 = length_prefix_message(s1) self.assertEquals(struct.unpack('>i', s2[0:4])[0], len(s1)) |