From 24031b7e82137d064fe69f995bd1a88162c7473b Mon Sep 17 00:00:00 2001 From: Jan Dittberner Date: Wed, 25 May 2011 16:06:42 +0200 Subject: [PATCH] fix one more test (addresses #112) --- migrate/tests/changeset/test_changeset.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/migrate/tests/changeset/test_changeset.py b/migrate/tests/changeset/test_changeset.py index 2b50922..d1b34cd 100644 --- a/migrate/tests/changeset/test_changeset.py +++ b/migrate/tests/changeset/test_changeset.py @@ -149,7 +149,10 @@ class TestAddDropColumn(fixture.DB): def add_func(col): return self.table.create_column(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) self.run_(add_func, drop_func)