diff options
Diffstat (limited to 'test/test_fetcher.py')
-rw-r--r-- | test/test_fetcher.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/test/test_fetcher.py b/test/test_fetcher.py index fea3f7d..984de88 100644 --- a/test/test_fetcher.py +++ b/test/test_fetcher.py @@ -6,11 +6,9 @@ import pytest from kafka.client_async import KafkaClient from kafka.consumer.fetcher import Fetcher from kafka.consumer.subscription_state import SubscriptionState -import kafka.errors as Errors -from kafka.future import Future from kafka.metrics import Metrics from kafka.protocol.fetch import FetchRequest -from kafka.structs import TopicPartition, OffsetAndMetadata +from kafka.structs import TopicPartition @pytest.fixture @@ -51,7 +49,7 @@ def test_send_fetches(fetcher, mocker): ] mocker.patch.object(fetcher, '_create_fetch_requests', - return_value = dict(enumerate(fetch_requests))) + return_value=dict(enumerate(fetch_requests))) ret = fetcher.send_fetches() for node, request in enumerate(fetch_requests): |