diff options
Diffstat (limited to 'sources/pyside6/tests/QtAsyncio/qasyncio_test_queues.py')
-rw-r--r-- | sources/pyside6/tests/QtAsyncio/qasyncio_test_queues.py | 70 |
1 files changed, 70 insertions, 0 deletions
diff --git a/sources/pyside6/tests/QtAsyncio/qasyncio_test_queues.py b/sources/pyside6/tests/QtAsyncio/qasyncio_test_queues.py new file mode 100644 index 000000000..0bd98c361 --- /dev/null +++ b/sources/pyside6/tests/QtAsyncio/qasyncio_test_queues.py @@ -0,0 +1,70 @@ +# Copyright (C) 2023 The Qt Company Ltd. +# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0 + +'''Test cases for QtAsyncio''' + +import unittest +import asyncio +import random +import time + +from PySide6.QtAsyncio import QAsyncioEventLoopPolicy + + +class QAsyncioTestCaseQueues(unittest.TestCase): + + async def produce(self, output, queue): + for _ in range(random.randint(0, 2)): + await asyncio.sleep(random.random()) + await queue.put(self.i) + output += f"{self.i} added to queue\n" + self.i += 1 + + async def consume(self, output, queue): + while True: + await asyncio.sleep(random.random()) + i = await queue.get() + output += f"{i} pulled from queue\n" + queue.task_done() + + async def main(self, output1, output2, num_producers, num_consumers): + self.i = 0 + queue = asyncio.Queue() + producers = [ + asyncio.create_task(self.produce(output1, queue)) for _ in range(num_producers)] + consumers = [ + asyncio.create_task(self.consume(output2, queue)) for _ in range(num_consumers)] + await asyncio.gather(*producers) + await queue.join() + for consumer in consumers: + consumer.cancel() + + def test_queues(self): + args = [(2, 3), (2, 1)] + for arg in args: + outputs_expected1 = [] + outputs_expected2 = [] + outputs_real1 = [] + outputs_real2 = [] + + # Run the code without QAsyncioEventLoopPolicy + random.seed(17) + start = time.perf_counter() + asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy()) + asyncio.run(self.main(outputs_expected1, outputs_expected2, *arg)) + end_expected = time.perf_counter() - start + + # Run the code with QAsyncioEventLoopPolicy and QtEventLoop + random.seed(17) + start = time.perf_counter() + asyncio.set_event_loop_policy(QAsyncioEventLoopPolicy()) + asyncio.run(self.main(outputs_real1, outputs_real2, *arg)) + end_real = time.perf_counter() - start + + self.assertEqual(outputs_expected1, outputs_real1) + self.assertEqual(outputs_expected2, outputs_real2) + self.assertAlmostEqual(end_expected, end_real, delta=1) + + +if __name__ == "__main__": + unittest.main() |