mirror of
https://github.com/soxoj/maigret.git
synced 2026-05-06 14:08:59 +00:00
Julio César Suástegui
c6cfef84ce
the <0.3/<0.4/etc upper bounds don't leave room for darwin or emulated/aarch64 runners, which have been seeing 0.7s+ on tests that expected <0.3s. bumped each upper bound by +0.7s. lower bounds unchanged — they still validate that tasks ran in parallel rather than serially. refs #679 Co-authored-by: Julio César Suástegui <juliosuas@users.noreply.github.com>
113 lines
4.2 KiB
Python
113 lines
4.2 KiB
Python
"""Maigret checking logic test functions"""
|
|
|
|
import pytest
|
|
import asyncio
|
|
import logging
|
|
from typing import Any, List, Tuple, Callable, Dict
|
|
from maigret.executors import (
|
|
AsyncioSimpleExecutor,
|
|
AsyncioProgressbarExecutor,
|
|
AsyncioProgressbarSemaphoreExecutor,
|
|
AsyncioProgressbarQueueExecutor,
|
|
AsyncioQueueGeneratorExecutor,
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def func(n):
|
|
await asyncio.sleep(0.1 * (n % 3))
|
|
return n
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_simple_asyncio_executor():
|
|
tasks: List[Tuple[Callable, list, dict]] = [(func, [n], {}) for n in range(10)]
|
|
executor = AsyncioSimpleExecutor(logger=logger)
|
|
assert await executor.run(tasks) == [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
|
|
assert executor.execution_time > 0.2
|
|
assert executor.execution_time < 1.0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_asyncio_progressbar_executor():
|
|
tasks: List[Tuple[Callable, list, dict]] = [(func, [n], {}) for n in range(10)]
|
|
|
|
executor = AsyncioProgressbarExecutor(logger=logger)
|
|
# no guarantees for the results order
|
|
assert sorted(await executor.run(tasks)) == [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
|
|
assert executor.execution_time > 0.2
|
|
assert executor.execution_time < 1.0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_asyncio_progressbar_semaphore_executor():
|
|
tasks: List[Tuple[Callable, list, dict]] = [(func, [n], {}) for n in range(10)]
|
|
|
|
executor = AsyncioProgressbarSemaphoreExecutor(logger=logger, in_parallel=5)
|
|
# no guarantees for the results order
|
|
assert sorted(await executor.run(tasks)) == [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
|
|
assert executor.execution_time > 0.2
|
|
assert executor.execution_time < 1.1
|
|
|
|
|
|
@pytest.mark.slow
|
|
@pytest.mark.asyncio
|
|
async def test_asyncio_progressbar_queue_executor():
|
|
tasks: List[Tuple[Callable, list, dict]] = [(func, [n], {}) for n in range(10)]
|
|
|
|
executor = AsyncioProgressbarQueueExecutor(logger=logger, in_parallel=2)
|
|
assert await executor.run(tasks) == [0, 1, 3, 2, 4, 6, 7, 5, 9, 8]
|
|
assert executor.execution_time > 0.5
|
|
assert executor.execution_time < 1.4
|
|
|
|
executor = AsyncioProgressbarQueueExecutor(logger=logger, in_parallel=3)
|
|
assert await executor.run(tasks) == [0, 3, 1, 4, 6, 2, 7, 9, 5, 8]
|
|
assert executor.execution_time > 0.4
|
|
assert executor.execution_time < 1.3
|
|
|
|
executor = AsyncioProgressbarQueueExecutor(logger=logger, in_parallel=5)
|
|
assert await executor.run(tasks) in (
|
|
[0, 3, 6, 1, 4, 7, 9, 2, 5, 8],
|
|
[0, 3, 6, 1, 4, 9, 7, 2, 5, 8],
|
|
)
|
|
assert executor.execution_time > 0.3
|
|
assert executor.execution_time < 1.2
|
|
|
|
executor = AsyncioProgressbarQueueExecutor(logger=logger, in_parallel=10)
|
|
assert await executor.run(tasks) == [0, 3, 6, 9, 1, 4, 7, 2, 5, 8]
|
|
assert executor.execution_time > 0.2
|
|
assert executor.execution_time < 1.1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_asyncio_queue_generator_executor():
|
|
tasks: List[Tuple[Callable, list, dict]] = [(func, [n], {}) for n in range(10)]
|
|
|
|
executor = AsyncioQueueGeneratorExecutor(logger=logger, in_parallel=2)
|
|
results = [result async for result in executor.run(tasks)] # type: ignore[arg-type]
|
|
assert results == [0, 1, 3, 2, 4, 6, 7, 5, 9, 8]
|
|
assert executor.execution_time > 0.5
|
|
assert executor.execution_time < 1.3
|
|
|
|
executor = AsyncioQueueGeneratorExecutor(logger=logger, in_parallel=3)
|
|
results = [result async for result in executor.run(tasks)] # type: ignore[arg-type]
|
|
assert results == [0, 3, 1, 4, 6, 2, 7, 9, 5, 8]
|
|
assert executor.execution_time > 0.4
|
|
assert executor.execution_time < 1.2
|
|
|
|
executor = AsyncioQueueGeneratorExecutor(logger=logger, in_parallel=5)
|
|
results = [result async for result in executor.run(tasks)] # type: ignore[arg-type]
|
|
assert results in (
|
|
[0, 3, 6, 1, 4, 7, 9, 2, 5, 8],
|
|
[0, 3, 6, 1, 4, 9, 7, 2, 5, 8],
|
|
)
|
|
assert executor.execution_time > 0.3
|
|
assert executor.execution_time < 1.1
|
|
|
|
executor = AsyncioQueueGeneratorExecutor(logger=logger, in_parallel=10)
|
|
results = [result async for result in executor.run(tasks)] # type: ignore[arg-type]
|
|
assert results == [0, 3, 6, 9, 1, 4, 7, 2, 5, 8]
|
|
assert executor.execution_time > 0.2
|
|
assert executor.execution_time < 1.0
|