diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/__init__.py | 0 | ||||
| -rw-r--r-- | test/integration.py | 2 | ||||
| -rw-r--r-- | test/unit.py | 4 |
3 files changed, 3 insertions, 3 deletions
diff --git a/test/__init__.py b/test/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/test/__init__.py diff --git a/test/integration.py b/test/integration.py index fccf28f..197f5ec 100644 --- a/test/integration.py +++ b/test/integration.py @@ -11,7 +11,7 @@ from threading import Thread, Event import time import unittest -from kafka import KafkaClient, ProduceRequest, FetchRequest +from kafka.kafka import KafkaClient, ProduceRequest, FetchRequest def get_open_port(): sock = socket.socket() 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 |