summaryrefslogtreecommitdiff
path: root/test/test_package.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_package.py')
-rw-r--r--test/test_package.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_package.py b/test/test_package.py
index a6a3a14..9b69a7c 100644
--- a/test/test_package.py
+++ b/test/test_package.py
@@ -1,6 +1,6 @@
-import unittest2
+from . import unittest
-class TestPackage(unittest2.TestCase):
+class TestPackage(unittest.TestCase):
def test_top_level_namespace(self):
import kafka as kafka1
self.assertEquals(kafka1.KafkaClient.__name__, "KafkaClient")