From 583d3ae22bf6fc373c03e50c7e4eb26eaa50db4b Mon Sep 17 00:00:00 2001 From: Mark Roberts Date: Wed, 23 Apr 2014 13:10:43 -0700 Subject: Fix Python 2.6 support --- test/test_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'test/test_client.py') diff --git a/test/test_client.py b/test/test_client.py index 9520d48..fe9beff 100644 --- a/test/test_client.py +++ b/test/test_client.py @@ -1,7 +1,7 @@ import os import random import struct -import unittest +import unittest2 from mock import MagicMock, patch @@ -15,7 +15,7 @@ from kafka.protocol import ( create_message, KafkaProtocol ) -class TestKafkaClient(unittest.TestCase): +class TestKafkaClient(unittest2.TestCase): def test_init_with_list(self): with patch.object(KafkaClient, 'load_metadata_for_topics'): client = KafkaClient(hosts=['kafka01:9092', 'kafka02:9092', 'kafka03:9092']) -- cgit v1.2.1