diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/integration.py | 2 | ||||
| -rw-r--r-- | test/unit.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/test/integration.py b/test/integration.py index 197f5ec..03e988e 100644 --- a/test/integration.py +++ b/test/integration.py @@ -11,7 +11,7 @@ from threading import Thread, Event import time import unittest -from kafka.kafka import KafkaClient, ProduceRequest, FetchRequest +from kafka.client import KafkaClient, ProduceRequest, FetchRequest def get_open_port(): sock = socket.socket() diff --git a/test/unit.py b/test/unit.py index 2db0857..6ca3564 100644 --- a/test/unit.py +++ b/test/unit.py @@ -3,7 +3,7 @@ import random import struct import unittest -from kafka.kafka import KafkaClient, ProduceRequest, FetchRequest, length_prefix_message +from kafka.client import KafkaClient, ProduceRequest, FetchRequest, length_prefix_message from kafka.codec import gzip_encode, gzip_decode ITERATIONS = 1000 |