~ubuntu-branches/debian/jessie/sqlalchemy/jessie

« back to all changes in this revision

Viewing changes to test/ext/test_compiler.py

  • Committer: Package Import Robot
  • Author(s): Piotr Ożarowski
  • Date: 2013-10-28 22:29:40 UTC
  • mfrom: (1.4.24)
  • Revision ID: package-import@ubuntu.com-20131028222940-wvyqffl4g617caun
Tags: 0.8.3-1
New upstream release

Show diffs side-by-side

added added

removed removed

Lines of Context:
4
4
                                    FunctionElement, Select, \
5
5
                                    BindParameter
6
6
 
7
 
from sqlalchemy.schema import DDLElement
 
7
from sqlalchemy.schema import DDLElement, CreateColumn, CreateTable
8
8
from sqlalchemy.ext.compiler import compiles, deregister
9
9
from sqlalchemy import exc
10
10
from sqlalchemy.sql import table, column, visitors
34
34
            "SELECT >>x<<, >>y<< WHERE >>MYTHINGY!<< = :MYTHINGY!_1"
35
35
        )
36
36
 
 
37
    def test_create_column_skip(self):
 
38
        @compiles(CreateColumn)
 
39
        def skip_xmin(element, compiler, **kw):
 
40
            if element.element.name == 'xmin':
 
41
                return None
 
42
            else:
 
43
                return compiler.visit_create_column(element, **kw)
 
44
 
 
45
        t = Table('t', MetaData(), Column('a', Integer),
 
46
                            Column('xmin', Integer),
 
47
                            Column('c', Integer))
 
48
 
 
49
        self.assert_compile(
 
50
            CreateTable(t),
 
51
            "CREATE TABLE t (a INTEGER, c INTEGER)"
 
52
        )
37
53
    def test_types(self):
38
54
        class MyType(TypeEngine):
39
55
            pass