Skip to content

Commit c5a6c63

Browse files
committed
Improved the test syntax and updated the Full import statement
1 parent 1cad32a commit c5a6c63

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

test/test_producer.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,10 @@
22

33
import logging
44

5+
try:
6+
from queue import Full
7+
except ImportError:
8+
from Queue import Full
59
from mock import MagicMock
610
from multiprocessing.queues import Full
711
from . import unittest
@@ -82,4 +86,4 @@ def test_producer_async_queue_normal(self):
8286
resp = producer.send_messages(topic, partition, *message_list)
8387
self.assertEqual(type(resp), list)
8488
self.assertEqual(producer.queue.qsize(), acceptable_size)
85-
self.assertEqual(producer.queue.full(), False)
89+
self.assertFalse(producer.queue.full())

0 commit comments

Comments
 (0)