Add test of types

Author: LuckySting

test/test_core.py

@@ -209,6 +209,21 @@ def test_select_types(self, connection):
         row = connection.execute(sa.select(tb)).fetchone()
         assert row == (1, "Hello World!", 3.5, True, now, today)
 
+    def test_integer_types(self, connection):
+        stmt = sa.Select(
+            sa.func.FormatType(sa.func.TypeOf(sa.bindparam("p_uint8", 8, types.UInt8))),
+            sa.func.FormatType(sa.func.TypeOf(sa.bindparam("p_uint16", 16, types.UInt16))),
+            sa.func.FormatType(sa.func.TypeOf(sa.bindparam("p_uint32", 32, types.UInt32))),
+            sa.func.FormatType(sa.func.TypeOf(sa.bindparam("p_uint64", 64, types.UInt64))),
+            sa.func.FormatType(sa.func.TypeOf(sa.bindparam("p_int8", -8, types.Int8))),
+            sa.func.FormatType(sa.func.TypeOf(sa.bindparam("p_int16", -16, types.Int16))),
+            sa.func.FormatType(sa.func.TypeOf(sa.bindparam("p_int32", -32, types.Int32))),
+            sa.func.FormatType(sa.func.TypeOf(sa.bindparam("p_int64", -64, types.Int64))),
+        )
+
+        result = connection.execute(stmt).fetchone()
+        assert result == (b"Uint8", b"Uint16", b"Uint32", b"Uint64", b"Int8", b"Int16", b"Int32", b"Int64")
+
 
 class TestWithClause(TablesTest):
     run_create_tables = "each"

ydb_sqlalchemy/sqlalchemy/__init__.py

@@ -99,16 +99,16 @@ def visit_uint16(self, type_: types.UInt16, **kw):
     def visit_uint8(self, type_: types.UInt8, **kw):
         return "UInt8"
 
-    def visit_int64(self, type_: types.UInt64, **kw):
+    def visit_int64(self, type_: types.Int64, **kw):
         return "Int64"
 
-    def visit_int32(self, type_: types.UInt32, **kw):
+    def visit_int32(self, type_: types.Int32, **kw):
         return "Int32"
 
-    def visit_int16(self, type_: types.UInt16, **kw):
+    def visit_int16(self, type_: types.Int16, **kw):
         return "Int16"
 
-    def visit_int8(self, type_: types.UInt8, **kw):
+    def visit_int8(self, type_: types.Int8, **kw):
         return "Int8"
 
     def visit_INTEGER(self, type_: sa.INTEGER, **kw):

ydb_sqlalchemy/sqlalchemy/types.py

@@ -12,7 +12,7 @@ class UInt32(types.Integer):
 
 
 class UInt16(types.Integer):
-    __visit_name__ = "uint32"
+    __visit_name__ = "uint16"
 
 
 class UInt8(types.Integer):