summaryrefslogtreecommitdiff
path: root/test/test_consumer_integration.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_consumer_integration.py')
-rw-r--r--test/test_consumer_integration.py8
1 files changed, 6 insertions, 2 deletions
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):