Python
/
ruff
mirror of https://github.com/astral-sh/ruff
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[ty] More SQLAlchemy test updates (#21846) 

Minor updates to the SQLAlchemy test suite. I verified all expected
results using pyright.
This commit is contained in:
David Peter 2025-12-08 15:22:55 +01:00 committed by GitHub
parent 4364ffbdd3
commit 4686111681
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
crates/ty_python_semantic/resources/mdtest/external/sqlalchemy.md vendored
View File

@ -45,9 +45,7 @@ invalid_user = User(invalid_arg=42)
First, set up a `Session`: First, set up a `Session`:
```py ```py
from datetime import datetime from sqlalchemy import select, Integer, Text, Boolean
from sqlalchemy import select, Integer, Text, Boolean, DateTime
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from sqlalchemy.orm import DeclarativeBase from sqlalchemy.orm import DeclarativeBase
from sqlalchemy.orm import Mapped, mapped_column from sqlalchemy.orm import Mapped, mapped_column
@ -126,15 +124,6 @@ for user_id, name in session.execute(stmt).tuples():
# TODO: should be `str` # TODO: should be `str`
reveal_type(name) # revealed: Unknown reveal_type(name) # revealed: Unknown
stmt = select(User.id, User.name).where(User.name == "Alice")
alice1 = session.scalars(stmt).first()
# TODO: should be `tuple[int, str] | None`
reveal_type(alice1) # revealed: Any | None
alice2 = session.scalar(stmt)
# TODO: should be `tuple[int, str] | None`
reveal_type(alice2) # revealed: Any
result = session.execute(stmt) result = session.execute(stmt)
row = result.one_or_none() row = result.one_or_none()
assert row is not None assert row is not None
@ -143,6 +132,19 @@ assert row is not None
reveal_type(user_id) # revealed: Unknown reveal_type(user_id) # revealed: Unknown
# TODO: should be `str` # TODO: should be `str`
reveal_type(name) # revealed: Unknown reveal_type(name) # revealed: Unknown
stmt = select(User.id).where(User.name == "Alice")
# TODO: should be `Select[tuple[int]]`
reveal_type(stmt) # revealed: Select[tuple[Unknown]]
alice_id = session.scalars(stmt).first()
# TODO: should be `int | None`
reveal_type(alice_id) # revealed: Unknown | None
alice_id = session.scalar(stmt)
# TODO: should be `int | None`
reveal_type(alice_id) # revealed: Unknown | None
``` ```
Using the legacy `query` API also works: Using the legacy `query` API also works: