diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_consumer.py | 3 | ||||
| -rw-r--r-- | test/test_consumer_integration.py | 3 |
2 files changed, 2 insertions, 4 deletions
diff --git a/test/test_consumer.py b/test/test_consumer.py index b33e537..7b8f370 100644 --- a/test/test_consumer.py +++ b/test/test_consumer.py @@ -2,8 +2,7 @@ from mock import MagicMock from . import unittest -from kafka.consumer import SimpleConsumer -from kafka.consumer.new import KafkaConsumer +from kafka import SimpleConsumer, KafkaConsumer from kafka.common import KafkaConfigurationError class TestKafkaConsumer(unittest.TestCase): diff --git a/test/test_consumer_integration.py b/test/test_consumer_integration.py index b4af70b..c97081b 100644 --- a/test/test_consumer_integration.py +++ b/test/test_consumer_integration.py @@ -3,12 +3,11 @@ import os from six.moves import xrange -from kafka import SimpleConsumer, MultiProcessConsumer, create_message +from kafka import SimpleConsumer, MultiProcessConsumer, KafkaConsumer, create_message from kafka.common import ( ProduceRequest, ConsumerFetchSizeTooSmall, ConsumerTimeout ) from kafka.consumer.base import MAX_FETCH_BUFFER_SIZE_BYTES -from kafka.consumer.new import KafkaConsumer from test.fixtures import ZookeeperFixture, KafkaFixture from test.testutil import ( |