def test_concurrent_producers_single_consumer(): num_threads = 100 num_items = 10 q = Queue() def producer(i): @gen.coroutine def run(): for j in range(num_items): yield q.put((i, j)) IOLoop(make_current=False).run_sync(run) producers = [ threading.Thread(target=producer, args=(i, )) for i in range(num_threads) ] for p in producers: p.start() items = defaultdict(lambda: []) for x in range(num_items * num_threads): i, j = yield q.get() items[i].append(j) for v in items.values(): assert v == list(range(num_items)) for p in producers: p.join()
def test_concurrent_producers_single_consumer(): num_threads = 100 num_items = 10 q = Queue() def producer(i): @gen.coroutine def run(): for j in range(num_items): yield q.put((i, j)) IOLoop(make_current=False).run_sync(run) producers = [ threading.Thread(target=producer, args=(i,)) for i in range(num_threads) ] for p in producers: p.start() items = defaultdict(lambda: []) for x in range(num_items * num_threads): i, j = yield q.get() items[i].append(j) for v in items.values(): assert v == list(range(num_items)) for p in producers: p.join()
def test_get_then_put(items): queue = Queue() got_futures = [] for i in range(len(items)): got_futures.append(queue.get()) for item in items: yield queue.put(item) got = yield got_futures assert got == items
def test_put_then_get_nowait(items): queue = Queue() for item in items: yield queue.put(item) got = [] for i in range(len(items)): got.append(queue.get_nowait()) assert got == items with pytest.raises(QueueEmpty): queue.get_nowait() future = queue.get() yield queue.put(42) assert 42 == (yield future)