From 519f535031f4b50b56842980a0b2988c7b959d42 Mon Sep 17 00:00:00 2001 From: Konsta Vesterinen Date: Wed, 3 Jul 2013 11:41:39 +0300 Subject: [PATCH] Added missing test case --- tests/test_tsvector_type.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tests/test_tsvector_type.py 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