forked from pool/python-kafka-python
17 lines
470 B
Diff
17 lines
470 B
Diff
|
Index: kafka-python-2.0.2/test/test_consumer_integration.py
|
||
|
|
===================================================================
|
||
|
|
--- kafka-python-2.0.2.orig/test/test_consumer_integration.py
|
||
|
|
+++ kafka-python-2.0.2/test/test_consumer_integration.py
|
||
|
|
@@ -1,7 +1,10 @@
|
||
|
|
import logging
|
||
|
|
import time
|
||
|
|
|
||
|
|
-from mock import patch
|
||
|
|
+try:
|
||
|
|
+ from unittest.mock import patch
|
||
|
|
+except ImportError:
|
||
|
|
+ from mock import patch
|
||
|
|
import pytest
|
||
|
|
from kafka.vendor.six.moves import range
|
||
|
|
|