Skip to content

Commit 712377a

Browse files
author
Dana Powers
committed
Import style nits in integration tests
1 parent ffcb49d commit 712377a

File tree

4 files changed

+10
-7
lines changed

4 files changed

+10
-7
lines changed

test/test_client_integration.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,8 @@
66
)
77

88
from test.fixtures import ZookeeperFixture, KafkaFixture
9-
from test.testutil import (
10-
KafkaIntegrationTestCase, kafka_versions
11-
)
9+
from test.testutil import KafkaIntegrationTestCase, kafka_versions
10+
1211

1312
class TestKafkaClientIntegration(KafkaIntegrationTestCase):
1413
@classmethod

test/test_consumer_integration.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,12 @@
33

44
from six.moves import xrange
55

6-
from kafka import SimpleConsumer, MultiProcessConsumer, KafkaConsumer, create_message
6+
from kafka import (
7+
KafkaConsumer, MultiProcessConsumer, SimpleConsumer, create_message
8+
)
79
from kafka.common import (
8-
ProduceRequest, ConsumerFetchSizeTooSmall, ConsumerTimeout, OffsetOutOfRangeError
10+
ProduceRequest, ConsumerFetchSizeTooSmall, ConsumerTimeout,
11+
OffsetOutOfRangeError
912
)
1013
from kafka.consumer.base import MAX_FETCH_BUFFER_SIZE_BYTES
1114

@@ -14,6 +17,7 @@
1417
KafkaIntegrationTestCase, kafka_versions, random_string, Timer
1518
)
1619

20+
1721
class TestConsumerIntegration(KafkaIntegrationTestCase):
1822
@classmethod
1923
def setUpClass(cls):

test/test_failover_integration.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,9 @@
44

55
from . import unittest
66

7-
from kafka import KafkaClient, SimpleConsumer
7+
from kafka import KafkaClient, SimpleConsumer, KeyedProducer
88
from kafka.common import TopicAndPartition, FailedPayloadsError, ConnectionError
99
from kafka.producer.base import Producer
10-
from kafka.producer import KeyedProducer
1110
from kafka.util import kafka_bytestring
1211

1312
from test.fixtures import ZookeeperFixture, KafkaFixture

test/test_producer_integration.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from test.fixtures import ZookeeperFixture, KafkaFixture
2020
from test.testutil import KafkaIntegrationTestCase, kafka_versions
2121

22+
2223
class TestKafkaProducerIntegration(KafkaIntegrationTestCase):
2324

2425
@classmethod

0 commit comments

Comments
 (0)