Add more cases

diff --git a/test/ext/mypy/plugin_files/issue_9156.py b/test/ext/mypy/plugin_files/issue_9156.py
index 0ae96b4f5c44304aece876b52098ec2cc59ac5c1..e00f2f12c62e6d8faebe6daee477f49bf6c6341d 100644 (file)
--- a/test/ext/mypy/plugin_files/issue_9156.py
+++ b/test/ext/mypy/plugin_files/issue_9156.py
@@ -3,7 +3,9 @@ from typing import Type, Any
 from sqlalchemy.sql.elements import ColumnElement
 from sqlalchemy.sql.type_api import TypeEngine
 
-col: ColumnElement
+col: ColumnElement[Any]
 type_: Type[TypeEngine[Any]]
+obj: TypeEngine[Any]
 
 col.cast(type_)
+col.cast(obj)