x/sqlalchemy-migrate
Code Issues Proposed changes

fix one more test (addresses #112)

Browse Source
This commit is contained in:
Jan Dittberner 2011-05-25 16:06:42 +02:00
parent e11a9e5cba
commit 24031b7e82
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
migrate/tests/changeset/test_changeset.py
View File

@ -149,7 +149,10 @@ class TestAddDropColumn(fixture.DB):
def add_func(col): def add_func(col):
return self.table.create_column(col) return self.table.create_column(col)
def drop_func(col): def drop_func(col):
self.table.c.remove(col) if SQLA_07:
self.table._columns.remove(col)
else:
self.table.c.remove(col)
return self.table.drop_column(col.name) return self.table.drop_column(col.name)
self.run_(add_func, drop_func) self.run_(add_func, drop_func)