From 109af6eeccafe51048ef82268582e67593b455c7 Mon Sep 17 00:00:00 2001
From: Yurii Karabas <1998uriyyo@gmail.com>
Date: Fri, 27 Jan 2023 14:31:30 +0200
Subject: [PATCH] Add more cases

---
 test/ext/mypy/plugin_files/issue_9156.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test/ext/mypy/plugin_files/issue_9156.py b/test/ext/mypy/plugin_files/issue_9156.py
index 0ae96b4f5c..e00f2f12c6 100644
--- 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)
-- 
2.47.2