Import style nits in integration tests

4 files changed, 10 insertions, 7 deletions

diff --git a/test/test_client_integration.py b/test/test_client_integration.py
index d963d85..585123b 100644
--- a/test/test_client_integration.py
+++ b/test/test_client_integration.py
@@ -6,9 +6,8 @@ from kafka.common import (
 )
 
 from test.fixtures import ZookeeperFixture, KafkaFixture
-from test.testutil import (
-    KafkaIntegrationTestCase, kafka_versions
-)
+from test.testutil import KafkaIntegrationTestCase, kafka_versions
+
 
 class TestKafkaClientIntegration(KafkaIntegrationTestCase):
     @classmethod
diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py
index 3825f94..c202c5c 100644
--- a/test/test_consumer_integration.py
+++ b/test/test_consumer_integration.py
@@ -3,9 +3,12 @@ import os
 
 from six.moves import xrange
 
-from kafka import SimpleConsumer, MultiProcessConsumer, KafkaConsumer, create_message
+from kafka import (
+    KafkaConsumer, MultiProcessConsumer, SimpleConsumer, create_message
+)
 from kafka.common import (
-    ProduceRequest, ConsumerFetchSizeTooSmall, ConsumerTimeout, OffsetOutOfRangeError
+    ProduceRequest, ConsumerFetchSizeTooSmall, ConsumerTimeout,
+    OffsetOutOfRangeError
 )
 from kafka.consumer.base import MAX_FETCH_BUFFER_SIZE_BYTES
 
@@ -14,6 +17,7 @@ from test.testutil import (
     KafkaIntegrationTestCase, kafka_versions, random_string, Timer
 )
 
+
 class TestConsumerIntegration(KafkaIntegrationTestCase):
     @classmethod
     def setUpClass(cls):
diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py
index 1d835e2..5082d7c 100644
--- a/test/test_failover_integration.py
+++ b/test/test_failover_integration.py
@@ -4,10 +4,9 @@ import time
 
 from . import unittest
 
-from kafka import KafkaClient, SimpleConsumer
+from kafka import KafkaClient, SimpleConsumer, KeyedProducer
 from kafka.common import TopicAndPartition, FailedPayloadsError, ConnectionError
 from kafka.producer.base import Producer
-from kafka.producer import KeyedProducer
 from kafka.util import kafka_bytestring
 
 from test.fixtures import ZookeeperFixture, KafkaFixture
diff --git a/test/test_producer_integration.py b/test/test_producer_integration.py
index 3439f4e..abf34c3 100644
--- a/test/test_producer_integration.py
+++ b/test/test_producer_integration.py
@@ -19,6 +19,7 @@ from kafka.producer.base import Producer
 from test.fixtures import ZookeeperFixture, KafkaFixture
 from test.testutil import KafkaIntegrationTestCase, kafka_versions
 
+
 class TestKafkaProducerIntegration(KafkaIntegrationTestCase):
 
     @classmethod