diff options
| author | Dana Powers <dana.powers@rd.io> | 2014-09-10 22:27:40 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2014-09-10 22:27:40 -0700 |
| commit | 04dbd0e7912c43e0d5cf32b29f0250dc67937df7 (patch) | |
| tree | d901f8be5eb72d04fdfd312b90d62fe9259157a5 /test/test_context.py | |
| parent | 2cfec9d1381b8e8d0667d704d89949fab3cf74dc (diff) | |
| download | kafka-python-04dbd0e7912c43e0d5cf32b29f0250dc67937df7.tar.gz | |
Fixup test_context.py unittest import
Diffstat (limited to 'test/test_context.py')
| -rw-r--r-- | test/test_context.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_context.py b/test/test_context.py index 8c3fd85..da9b22f 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.common import OffsetOutOfRangeError from kafka.context import OffsetCommitContext -class TestOffsetCommitContext(TestCase): +class TestOffsetCommitContext(unittest.TestCase): """ OffsetCommitContext tests. """ |