From 83af5102e995e854a1980b90f1400afdd098da37 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bruno=20Reni=C3=A9?= Date: Thu, 28 Aug 2014 14:40:01 +0200 Subject: Use base unittest or unittest2 depending on python version --- test/testutil.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'test/testutil.py') diff --git a/test/testutil.py b/test/testutil.py index dc8eea0..e7dcda6 100644 --- a/test/testutil.py +++ b/test/testutil.py @@ -5,9 +5,10 @@ import random import socket import string import time -import unittest2 import uuid +from . import unittest + from kafka.common import OffsetRequest from kafka import KafkaClient @@ -45,7 +46,7 @@ def get_open_port(): sock.close() return port -class KafkaIntegrationTestCase(unittest2.TestCase): +class KafkaIntegrationTestCase(unittest.TestCase): create_client = True topic = None server = None -- cgit v1.2.1