diff --git a/tests/test_tsvector_type.py b/tests/test_tsvector_type.py new file mode 100644 index 0000000..b6b4b76 --- /dev/null +++ b/tests/test_tsvector_type.py @@ -0,0 +1,20 @@ +import sqlalchemy as sa +from sqlalchemy_utils import TSVectorType +from tests import TestCase + + +class TestTSVector(TestCase): + def create_models(self): + class User(self.Base): + __tablename__ = 'user' + id = sa.Column(sa.Integer, primary_key=True) + name = sa.Column(sa.Unicode(255)) + search_index = sa.Column(TSVectorType()) + + def __repr__(self): + return 'User(%r)' % self.id + + self.User = User + + def test_generates_table(self): + assert 'search_index' in self.User.__table__.c