summaryrefslogtreecommitdiff
path: root/test/testutil.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/testutil.py')
-rw-r--r--test/testutil.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/testutil.py b/test/testutil.py
index 61fe9bd..78e6f7d 100644
--- a/test/testutil.py
+++ b/test/testutil.py
@@ -5,7 +5,7 @@ import random
import socket
import string
import time
-import unittest
+import unittest2
import uuid
from kafka.common import OffsetRequest
@@ -56,7 +56,7 @@ def get_open_port():
sock.close()
return port
-class KafkaIntegrationTestCase(unittest.TestCase):
+class KafkaIntegrationTestCase(unittest2.TestCase):
create_client = True
topic = None