scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(timeout): test timeout extension against regressions with delayed call

Browse source 
Signed-off-by: agustin18 <agustinsaiz02@gmail.com>
This commit is contained in:
agustin18 2026-09-28 20:04:59 +00:00
parent b23e6585b8
commit 3447c3529a
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/unit/test_timeout.py
View file

@ -25,12 +25,16 @@ async def test_async_timeout_decorator_enforces_per_call_timeout(timeout_arg: st
@pytest.mark.asyncio
async def test_async_timeout_decorator_extends_duration_with_per_call_timeout() -> None:
@pytest.mark.parametrize("timeout_arg", ["request_timeout", "force_timeout"])
async def test_async_timeout_decorator_extends_duration_with_per_call_timeout(
timeout_arg: str,
) -> None:
@timeout(timeout_duration=0.001)
async def fast_func(**kwargs):
async def delayed_func(**kwargs):
await asyncio.sleep(0.01)
return "completed"
result: Final = await fast_func(request_timeout=60.0, model="test-model")
result: Final = await delayed_func(**{timeout_arg: 60.0, "model": "test-model"})
assert result == "completed"