Fixup test_context.py unittest import

Dana Powers · Dana Powers · commit 04dbd0e7912c · 2014-09-10T22:27:40.000-07:00
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.
     """
