diff options
Diffstat (limited to 'test/test_conn.py')
| -rw-r--r-- | test/test_conn.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_conn.py b/test/test_conn.py index 5451398..931ace7 100644 --- a/test/test_conn.py +++ b/test/test_conn.py @@ -4,8 +4,8 @@ import struct import mock import unittest2 -from kafka.common import * -from kafka.conn import * +from kafka.common import ConnectionError +from kafka.conn import KafkaConnection, collect_hosts, DEFAULT_SOCKET_TIMEOUT_SECONDS class ConnTest(unittest2.TestCase): def setUp(self): |