diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_autogen_diffs.py | 49 | ||||
| -rw-r--r-- | tests/test_postgresql.py | 6 |
2 files changed, 31 insertions, 24 deletions
diff --git a/tests/test_autogen_diffs.py b/tests/test_autogen_diffs.py index 196ba23..d176b91 100644 --- a/tests/test_autogen_diffs.py +++ b/tests/test_autogen_diffs.py @@ -396,21 +396,23 @@ class AutogenerateDiffTest(ModelOne, AutogenTest, TestBase): def test_skip_null_type_comparison_reflected(self): ac = ops.AlterColumnOp("sometable", "somecol") - autogenerate.compare._compare_type(None, "sometable", "somecol", - Column("somecol", NULLTYPE), - Column("somecol", Integer()), - ac, self.autogen_context - ) + autogenerate.compare._compare_type( + self.autogen_context, ac, + None, "sometable", "somecol", + Column("somecol", NULLTYPE), + Column("somecol", Integer()), + ) diff = ac.to_diff_tuple() assert not diff def test_skip_null_type_comparison_local(self): ac = ops.AlterColumnOp("sometable", "somecol") - autogenerate.compare._compare_type(None, "sometable", "somecol", - Column("somecol", Integer()), - Column("somecol", NULLTYPE), - ac, self.autogen_context - ) + autogenerate.compare._compare_type( + self.autogen_context, ac, + None, "sometable", "somecol", + Column("somecol", Integer()), + Column("somecol", NULLTYPE), + ) diff = ac.to_diff_tuple() assert not diff @@ -422,19 +424,22 @@ class AutogenerateDiffTest(ModelOne, AutogenTest, TestBase): return isinstance(conn_type, Integer) ac = ops.AlterColumnOp("sometable", "somecol") - autogenerate.compare._compare_type(None, "sometable", "somecol", - Column("somecol", INTEGER()), - Column("somecol", MyType()), - ac, self.autogen_context - ) + autogenerate.compare._compare_type( + self.autogen_context, ac, + None, "sometable", "somecol", + Column("somecol", INTEGER()), + Column("somecol", MyType()), + ) + assert not ac.has_changes() ac = ops.AlterColumnOp("sometable", "somecol") - autogenerate.compare._compare_type(None, "sometable", "somecol", - Column("somecol", String()), - Column("somecol", MyType()), - ac, self.autogen_context - ) + autogenerate.compare._compare_type( + self.autogen_context, ac, + None, "sometable", "somecol", + Column("somecol", String()), + Column("somecol", MyType()), + ) diff = ac.to_diff_tuple() eq_( diff[0][0:4], @@ -453,10 +458,10 @@ class AutogenerateDiffTest(ModelOne, AutogenTest, TestBase): uo = ops.AlterColumnOp('sometable', 'somecol') autogenerate.compare._compare_type( + self.autogen_context, uo, None, "sometable", "somecol", Column("somecol", Integer, nullable=True), - Column("somecol", MyType()), - uo, self.autogen_context + Column("somecol", MyType()) ) assert not uo.has_changes() diff --git a/tests/test_postgresql.py b/tests/test_postgresql.py index 7fb514f..576d957 100644 --- a/tests/test_postgresql.py +++ b/tests/test_postgresql.py @@ -203,8 +203,10 @@ class PostgresqlDefaultCompareTest(TestBase): insp_col = Column("somecol", cols[0]['type'], server_default=text(cols[0]['default'])) op = ops.AlterColumnOp("test", "somecol") - _compare_server_default(None, "test", "somecol", insp_col, - t2.c.somecol, op, self.autogen_context) + _compare_server_default( + self.autogen_context, op, + None, "test", "somecol", insp_col, t2.c.somecol) + diffs = op.to_diff_tuple() eq_(bool(diffs), diff_expected) |