def test_except_with_std_queue_empty(self):
q = Queue(1)
with self.assertRaises(std_queue.Empty):
q.get(block=False)
with self.assertRaises(Empty):
q.get(block=False)
def test_unrealistic_timeout(self):
q = Queue()
with self.assertRaises(OverflowError):
q.get(True, 2 ** 72)
with self.assertRaises(OverflowError):
q.put(1, True, 2 ** 72)
def test_negative_timeout(self):
q = Queue()
with self.assertRaises(ValueError):
q.get(True, -1)
with self.assertRaises(ValueError):
q.put('data', True, -1)
def test_get_many_not_enough_space(self):
q = Queue(10)
msg = "you want to get 12 but maxsize is 10"
with self.assertRaisesRegexp(ValueError, msg):
q.get_many(12)
with self.assertRaises(Empty):
q.get_many(2, block=False)
def test_get_many_enough_space(self):
q = Queue(10)
q.put_many((1, 2, 3))
q.put_many((1, 2, 3))
self.assertEqual((1, 2), q.get_many(2))
self.assertEqual((3, 1), q.get_many(2))
self.assertEqual((2, 3), q.get_many(2))
def test_except_with_std_queue_full(self):
q = Queue(1)
q.put(1)
with self.assertRaises(std_queue.Full):
q.put(1, block=False)
with self.assertRaises(Full):
q.put(1, block=False)
def test_put_many_enough_space(self):
q = Queue(10)
q.put_many((1, 2, 3))
self.assertEqual(3, q.qsize())
self.assertEqual(1, q.get())
self.assertEqual(2, q.get())
self.assertEqual(3, q.get())
def test_put_many_too_many_items(self):
q = Queue(1)
msg = "items of size 3 is bigger then maxsize: 1"
with self.assertRaisesRegexp(ValueError, msg):
q.put_many((1, 2, 3))
q.put(None)
with self.assertRaises(Full):
q.put_many([1], block=False)
def test_get_put(self):
q = Queue(2)
q.put(1, 1)
q.put(2, 1)
with self.assertRaises(Full):
q.put(None, 1, 1)
self.assertEqual(1, q.get())
self.assertEqual(2, q.get())
with self.assertRaises(Empty):
q.get(1, 1)
def test_unrealistic_timeout(self):
q = Queue()
with self.assertRaises(OverflowError):
q.get(True, 2**72)
with self.assertRaises(OverflowError):
q.put(1, True, 2**72)
def test_negative_timeout(self):
q = Queue()
with self.assertRaises(ValueError):
q.get(True, -1)
with self.assertRaises(ValueError):
q.put('data', True, -1)
def test_except_with_std_queue_empty(self):
q = Queue(1)
with self.assertRaises(std_queue.Empty):
q.get(block=False)
with self.assertRaises(Empty):
q.get(block=False)
def test_get_many_enough_space(self):
q = Queue(10)
q.put_many((1, 2, 3))
q.put_many((1, 2, 3))
self.assertEqual((1, 2), q.get_many(2))
self.assertEqual((3, 1), q.get_many(2))
self.assertEqual((2, 3), q.get_many(2))
def test_put_many_enough_space(self):
q = Queue(10)
q.put_many((1, 2, 3))
self.assertEqual(3, q.qsize())
self.assertEqual(1, q.get())
self.assertEqual(2, q.get())
self.assertEqual(3, q.get())
def test_get_many_not_enough_space(self):
q = Queue(10)
msg = "you want to get 12 but maxsize is 10"
with self.assertRaisesRegexp(ValueError, msg):
q.get_many(12)
with self.assertRaises(Empty):
q.get_many(2, block=False)
def test_get_put_with_thread_and_late_get(self):
def producer(q):
[q.put(x, True, 0.1) for x in range(400)]
def consumer(test, q):
for x in range(400):
self.assertEqual(x, q.get(True, 0.1))
queue = Queue(400)
t1 = threading.Thread(target=producer, args=(queue, ))
t1.start()
time.sleep(0.2)
t2 = threading.Thread(target=consumer, args=(self, queue))
t2.start()
t1.join()
t2.join()
def test_get_put_with_thread_and_late_put(self):
def consumer(test, q):
to_consume = range(40)
for x in range(40):
to_consume.remove(q.get(True, 4))
self.assertEqual(to_consume, [])
def producer(q):
for x in range(40):
q.put(x, True, 0.1)
queue = Queue(40)
t1 = threading.Thread(target=consumer, args=(self, queue))
t1.start()
time.sleep(1)
t2 = threading.Thread(target=producer, args=(queue, ))
t2.start()
t1.join()
t2.join()
def test_negative_max_size(self):
q = Queue(-1000)
self.assertEqual(q.maxsize, 0)
def test_unrealistic_max_size(self):
with self.assertRaises(OverflowError):
Queue(2**72)
with self.assertRaises(OverflowError):
Queue(-2**72)
def test_get_with_empty_error(self):
q = Queue(1)
with self.assertRaises(Empty):
q.get(1, 0.1)
def test_put_with_full_error(self):
q = Queue(1)
q.put(None)
with self.assertRaises(Full):
q.put(None, True, 2.1)
def test_put_with_full_error(self):
q = Queue(1)
q.put(None)
with self.assertRaises(Full):
q.put(None, True, 2.1)
def test_put(self):
q = Queue(1)
q.put(None)
def test_put(self):
q = Queue(1)
q.put(None)
def test_maxsize_get(self):
q = Queue(100)
self.assertEqual(q.maxsize, 100)
self.c = {}
class Con(threading.Thread):
def __init__(self, q):
self.q = q
threading.Thread.__init__(self)
def run(self):
for _ in xrange(10**6):
self.q.get()
class Prod(threading.Thread):
def __init__(self, q):
self.q = q
threading.Thread.__init__(self)
def run(self):
for _ in xrange(10**6):
self.q.put(BigFatObject())
if __name__ == '__main__':
q = Queue()
nb_threads = 1
cons = [Con(q) for _ in range(nb_threads)]
prods = [Prod(q) for _ in range(nb_threads)]
start = time.time()
[x.start() for x in cons + prods]
[x.join() for x in prods + cons]
print time.time() - start
def test_get_with_empty_error(self):
q = Queue(1)
with self.assertRaises(Empty):
q.get(1, 0.1)
