Test fixes after flake8 run

Author: mrtheb

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
 

test/test_unit.py

@@ -3,7 +3,8 @@
 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))