fix test error

2020-09-21 15:03:16 +08:00
parent dc821d8a02
commit 4b0c4ae7d0
2 changed files with 4 additions and 0 deletions
--- a/tests/test_ddl.py
+++ b/tests/test_ddl.py
@@ -136,6 +136,8 @@ def test_drop_column():
    if isinstance(Migrate.ddl, SqliteDDL):
        with pytest.raises(NotSupportError):
            ret = Migrate.ddl.drop_column(Category, "name")
+    else:
+        ret = Migrate.ddl.drop_column(Category, "name")
    if isinstance(Migrate.ddl, MysqlDDL):
        assert ret == "ALTER TABLE `category` DROP COLUMN `name`"
    elif isinstance(Migrate.ddl, PostgresDDL):