add tests for SmallInt and BigInt

Author: dantownsend

piccolo/columns/base.py

@@ -359,7 +359,7 @@ def _validate_default(
         ):
             self._validated = True
             return True
-        elif inspect.isfunction(default):
+        elif callable(default):
             # We need to prevent recursion, otherwise a function which returns
             # a function would be an infinite loop.
             if allow_recursion and self._validate_default(

tests/apps/migrations/auto/integration/test_migrations.py

@@ -9,7 +9,14 @@
 
 from piccolo.columns.defaults.uuid import UUID4
 from piccolo.conf.apps import AppConfig
-from piccolo.columns.column_types import Integer, Text, UUID, Varchar
+from piccolo.columns.column_types import (
+    BigInt,
+    Integer,
+    SmallInt,
+    Text,
+    UUID,
+    Varchar,
+)
 from piccolo.apps.migrations.commands.new import (
     _create_new_migration,
     _create_migrations_folder,
@@ -137,6 +144,38 @@ def test_integer_column(self):
             ]
         )
 
+    def test_smallint_column(self):
+        self._test_migrations(
+            table_classes=[
+                self.table(column)
+                for column in [
+                    SmallInt(),
+                    SmallInt(default=1),
+                    SmallInt(default=integer_default),
+                    SmallInt(null=True),
+                    SmallInt(null=False),
+                    SmallInt(index=True),
+                    SmallInt(index=False),
+                ]
+            ]
+        )
+
+    def test_bigint_column(self):
+        self._test_migrations(
+            table_classes=[
+                self.table(column)
+                for column in [
+                    BigInt(),
+                    BigInt(default=1),
+                    BigInt(default=integer_default),
+                    BigInt(null=True),
+                    BigInt(null=False),
+                    BigInt(index=True),
+                    BigInt(index=False),
+                ]
+            ]
+        )
+
     def test_uuid_column(self):
         self._test_migrations(
             table_classes=[