diff options
Diffstat (limited to 'test/test_consumer_integration.py')
| -rw-r--r-- | test/test_consumer_integration.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py index 1b60c95..4e081ce 100644 --- a/test/test_consumer_integration.py +++ b/test/test_consumer_integration.py @@ -7,11 +7,9 @@ from . import unittest from kafka import ( KafkaConsumer, MultiProcessConsumer, SimpleConsumer, create_message ) -from kafka.common import ( - ProduceRequestPayload, ConsumerFetchSizeTooSmall, - OffsetOutOfRangeError, TopicPartition -) from kafka.consumer.base import MAX_FETCH_BUFFER_SIZE_BYTES +from kafka.errors import ConsumerFetchSizeTooSmall, OffsetOutOfRangeError +from kafka.structs import ProduceRequestPayload, TopicPartition from test.fixtures import ZookeeperFixture, KafkaFixture from test.testutil import ( |