summaryrefslogtreecommitdiff
path: root/test/unit.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/unit.py')
-rw-r--r--test/unit.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/unit.py b/test/unit.py
index 5c48f27..c77aa81 100644
--- a/test/unit.py
+++ b/test/unit.py
@@ -3,8 +3,8 @@ import random
import struct
import unittest
-from kafka import KafkaClient, ProduceRequest, FetchRequest
-from kafka import gzip_encode, gzip_decode, length_prefix_message
+from kafka.kafka import KafkaClient, ProduceRequest, FetchRequest
+from kafka.kafka import gzip_encode, gzip_decode, length_prefix_message
ITERATIONS = 1000
STRLEN = 100