diff options
Diffstat (limited to 'test/test_producer.py')
| -rw-r--r-- | test/test_producer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index a84e20f..9d21369 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -4,14 +4,14 @@ import logging import os import random import struct -import unittest2 +from . import unittest from mock import MagicMock, patch from kafka import KafkaClient from kafka.producer import Producer -class TestKafkaProducer(unittest2.TestCase): +class TestKafkaProducer(unittest.TestCase): def test_producer_message_types(self): producer = Producer(MagicMock()) |