diff options
Diffstat (limited to 'test/test_consumer.py')
| -rw-r--r-- | test/test_consumer.py | 3 |
1 files changed, 1 insertions, 2 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): |