From 04dbd0e7912c43e0d5cf32b29f0250dc67937df7 Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@rd.io>
Date: Wed, 10 Sep 2014 22:27:40 -0700
Subject: [PATCH] Fixup test_context.py unittest import

---
 test/test_context.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/test_context.py b/test/test_context.py
index 8c3fd856d..da9b22f65 100644
--- a/test/test_context.py
+++ b/test/test_context.py
@@ -1,7 +1,7 @@
 """
 OffsetCommitContext tests.
 """
-from unittest2 import TestCase
+from . import unittest
 
 from mock import MagicMock, patch
 
@@ -9,7 +9,7 @@
 from kafka.context import OffsetCommitContext
 
 
-class TestOffsetCommitContext(TestCase):
+class TestOffsetCommitContext(unittest.TestCase):
     """
     OffsetCommitContext tests.
     """