- Fixed regression from 0.6 where SMALLINT and

BIGINT types would both generate SERIAL
on an integer PK column, instead of
SMALLINT and BIGSERIAL [ticket:2065]

1 files changed, 8 insertions, 0 deletions

diff --git a/lib/sqlalchemy/types.py b/lib/sqlalchemy/types.py
index a46cb1530..71b55f33f 100644
--- a/lib/sqlalchemy/types.py
+++ b/lib/sqlalchemy/types.py
@@ -1007,6 +1007,10 @@ class SmallInteger(Integer):
 
     __visit_name__ = 'small_integer'
 
+    @property
+    def _type_affinity(self):
+        return SmallInteger
+
 class BigInteger(Integer):
     """A type for bigger ``int`` integers.
 
@@ -1017,6 +1021,10 @@ class BigInteger(Integer):
 
     __visit_name__ = 'big_integer'
 
+    @property
+    def _type_affinity(self):
+        return BigInteger
+
 class Numeric(_DateAffinity, TypeEngine):
     """A type for fixed precision numbers.