diff options
Diffstat (limited to 'test/test_failover_integration.py')
| -rw-r--r-- | test/test_failover_integration.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py index 9409241..58e9463 100644 --- a/test/test_failover_integration.py +++ b/test/test_failover_integration.py @@ -3,10 +3,10 @@ import os import time from kafka import SimpleClient, SimpleConsumer, KeyedProducer -from kafka.common import ( - TopicPartition, FailedPayloadsError, ConnectionError, RequestTimedOutError -) +from kafka.errors import ( + FailedPayloadsError, ConnectionError, RequestTimedOutError) from kafka.producer.base import Producer +from kafka.structs import TopicPartition from test.fixtures import ZookeeperFixture, KafkaFixture from test.testutil import KafkaIntegrationTestCase, random_string |