fix(mysql): add dtype mapping for mediumint

ibis-project · Sep 13, 2024 · 6d22c5c · 6d22c5c
1 parent a1bd3ba
commit 6d22c5c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/ibis/backends/mysql/tests/test_client.py b/ibis/backends/mysql/tests/test_client.py
@@ -27,8 +27,8 @@
     param("boolean", dt.int8, id="boolean"),
     param("smallint", dt.int16, id="smallint"),
     param("int2", dt.int16, id="int2"),
-    # ("mediumint", dt.int32), => https://github.com/tobymao/sqlglot/issues/2109
-    # ("int3", dt.int32), => https://github.com/tobymao/sqlglot/issues/2109
+    param("mediumint", dt.int32, id="mediumint"),
+    param("int3", dt.int32, id="int3"),
     param("int", dt.int32, id="int"),
     param("int4", dt.int32, id="int4"),
     param("integer", dt.int32, id="integer"),

diff --git a/ibis/backends/sql/datatypes.py b/ibis/backends/sql/datatypes.py
@@ -40,6 +40,7 @@
     typecode.LONGBLOB: dt.Binary,
     typecode.LONGTEXT: dt.String,
     typecode.MEDIUMBLOB: dt.Binary,
+    typecode.MEDIUMINT: dt.Int32,
     typecode.MEDIUMTEXT: dt.String,
     typecode.MONEY: dt.Decimal(19, 4),
     typecode.NCHAR: dt.String,