diff options
Diffstat (limited to 'test/test_consumer_integration.py')
-rw-r--r-- | test/test_consumer_integration.py | 8 |
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): |