summaryrefslogtreecommitdiff
path: root/test/ext/mypy/plain_files/dml.py
diff options
context:
space:
mode:
authorFederico Caselli <cfederico87@gmail.com>2023-03-27 22:03:44 +0200
committerMike Bayer <mike_mp@zzzcomputing.com>2023-03-27 19:46:09 -0400
commit2dc24ced0e94c50623aab41c75e42883d68a846a (patch)
tree30976c74974e49e60a5917cff8d547f2a2b46003 /test/ext/mypy/plain_files/dml.py
parentc7ce7ff0225fe0ddaf63f0706429b885410de365 (diff)
downloadsqlalchemy-2dc24ced0e94c50623aab41c75e42883d68a846a.tar.gz
Improve from_select typing
Properly type :paramref:`_dml.Insert.from_select.names` to accept a list of string or columns or mapped attributes. Fixes: #9514 Change-Id: I2a3396454e79bd2113fc3d57d1252a2e6d141310
Diffstat (limited to 'test/ext/mypy/plain_files/dml.py')
-rw-r--r--test/ext/mypy/plain_files/dml.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/test/ext/mypy/plain_files/dml.py b/test/ext/mypy/plain_files/dml.py
index d2ffbf1e1..596f47795 100644
--- a/test/ext/mypy/plain_files/dml.py
+++ b/test/ext/mypy/plain_files/dml.py
@@ -5,6 +5,7 @@ from typing import Dict
from sqlalchemy import Column
from sqlalchemy import insert
+from sqlalchemy import select
from sqlalchemy.orm import DeclarativeBase
from sqlalchemy.orm import Mapped
from sqlalchemy.orm import mapped_column
@@ -33,3 +34,8 @@ stmt2 = insert(User).values(d2)
d3: Dict[Column[str], Any] = {}
stmt3 = insert(User).values(d3)
+
+stmt4 = insert(User).from_select(
+ [User.id, "name", User.__table__.c.data],
+ select(User.id, User.name, User.data),
+)