Fix LowCardinality and Nullable alembic generation xzkostyan#217

Author: xzkostyan

clickhouse_sqlalchemy/types/common.py

@@ -52,7 +52,7 @@ class Nullable(ClickHouseTypeEngine):
     __visit_name__ = 'nullable'
 
     def __init__(self, nested_type):
-        self.nested_type = nested_type
+        self.nested_type = to_instance(nested_type)
         super(Nullable, self).__init__()
 
 
@@ -64,7 +64,7 @@ class LowCardinality(ClickHouseTypeEngine):
     __visit_name__ = 'lowcardinality'
 
     def __init__(self, nested_type):
-        self.nested_type = nested_type
+        self.nested_type = to_instance(nested_type)
         super(LowCardinality, self).__init__()
 
 

tests/test_reflection.py

@@ -62,7 +62,7 @@ def test_nullable(self):
 
         self.assertIsInstance(col['type'], types.Nullable)
         self.assertTrue(col['nullable'])
-        self.assertEqual(col['type'].nested_type, types.Int32)
+        self.assertIsInstance(col['type'].nested_type, types.Int32)
 
     def test_not_null(self):
         metadata = self.metadata()
@@ -84,7 +84,7 @@ def test_low_cardinality(self):
         )[0]['type']
 
         self.assertIsInstance(coltype, types.LowCardinality)
-        self.assertEqual(coltype.nested_type, types.String)
+        self.assertIsInstance(coltype.nested_type, types.String)
 
     def test_tuple(self):
         coltype = self._type_round_trip(