diff options
Diffstat (limited to 'test/test_consumer.py')
| -rw-r--r-- | test/test_consumer.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py index e664292..f3dad16 100644 --- a/test/test_consumer.py +++ b/test/test_consumer.py @@ -4,11 +4,11 @@ from mock import MagicMock, patch from . import unittest from kafka import SimpleConsumer, KafkaConsumer, MultiProcessConsumer -from kafka.common import ( - KafkaConfigurationError, FetchResponsePayload, OffsetFetchResponsePayload, - FailedPayloadsError, OffsetAndMessage, - NotLeaderForPartitionError, UnknownTopicOrPartitionError -) +from kafka.errors import ( + FailedPayloadsError, KafkaConfigurationError, NotLeaderForPartitionError, + UnknownTopicOrPartitionError) +from kafka.structs import ( + FetchResponsePayload, OffsetAndMessage, OffsetFetchResponsePayload) class TestKafkaConsumer(unittest.TestCase): |