Fixup test_context.py unittest import

quard8 · Sep 11, 2014 · 04dbd0e · 04dbd0e
1 parent 2cfec9d
commit 04dbd0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_context.py b/test/test_context.py
@@ -1,15 +1,15 @@
 """
 OffsetCommitContext tests.
 """
-from unittest2 import TestCase
+from . import unittest
 
 from mock import MagicMock, patch
 
 from kafka.common import OffsetOutOfRangeError
 from kafka.context import OffsetCommitContext
 
 
-class TestOffsetCommitContext(TestCase):
+class TestOffsetCommitContext(unittest.TestCase):
     """
     OffsetCommitContext tests.
     """