Update imports from kafka.common -> kafka.errors / kafka.structs

1 files changed, 4 insertions, 7 deletions

diff --git a/test/test_client.py b/test/test_client.py
index 6980434..42d7dbd 100644
--- a/test/test_client.py
+++ b/test/test_client.py
@@ -6,17 +6,14 @@ import six
 from . import unittest
 
 from kafka import SimpleClient
-from kafka.common import (
-    ProduceRequestPayload,
-    BrokerMetadata,
-    TopicPartition, KafkaUnavailableError,
-    LeaderNotAvailableError, UnknownTopicOrPartitionError,
-    KafkaTimeoutError, ConnectionError, FailedPayloadsError
-)
 from kafka.conn import KafkaConnection
+from kafka.errors import (
+    KafkaUnavailableError, LeaderNotAvailableError, KafkaTimeoutError,
+    UnknownTopicOrPartitionError, ConnectionError, FailedPayloadsError)
 from kafka.future import Future
 from kafka.protocol import KafkaProtocol, create_message
 from kafka.protocol.metadata import MetadataResponse
+from kafka.structs import ProduceRequestPayload, BrokerMetadata, TopicPartition
 
 from test.testutil import Timer