From 07efdf8887eb96f868e27c645ce9dfdc2dec350c Mon Sep 17 00:00:00 2001 From: David Peter Date: Mon, 8 Dec 2025 12:59:28 +0100 Subject: [PATCH] [ty] Test for @asynccontextmanager --- .../resources/mdtest/async.md | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/crates/ty_python_semantic/resources/mdtest/async.md b/crates/ty_python_semantic/resources/mdtest/async.md index 416c88b09c..1b9b16f827 100644 --- a/crates/ty_python_semantic/resources/mdtest/async.md +++ b/crates/ty_python_semantic/resources/mdtest/async.md @@ -82,6 +82,24 @@ async def main(): reveal_type(b) # revealed: int ``` +### `asynccontextmanager` + +```py +from contextlib import asynccontextmanager +from typing import AsyncGenerator + +class Session: ... + +@asynccontextmanager +async def connect() -> AsyncGenerator[Session]: + yield Session() + +async def main(): + async with connect() as session: + # TODO: should be `Session` + reveal_type(session) # revealed: Unknown +``` + ## Under the hood ```toml