mirror of https://github.com/astral-sh/ruff
[ty] Test for @asynccontextmanager
This commit is contained in:
parent
dfd6ed0524
commit
07efdf8887
|
|
@ -82,6 +82,24 @@ async def main():
|
||||||
reveal_type(b) # revealed: int
|
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
|
## Under the hood
|
||||||
|
|
||||||
```toml
|
```toml
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue