From ad913f242d6b3de3d188d5a28977e33b14bcda16 Mon Sep 17 00:00:00 2001 From: Michael Manfre Date: Sun, 2 Feb 2014 15:44:01 -0500 Subject: [PATCH] Fix regress added to migrations.test_operations.test_alter_field_pk_fk --- tests/migrations/test_operations.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/migrations/test_operations.py b/tests/migrations/test_operations.py index 375a9ccc54c..01f78b62b5d 100644 --- a/tests/migrations/test_operations.py +++ b/tests/migrations/test_operations.py @@ -349,11 +349,12 @@ class OperationTests(MigrationTestBase): self.assertIsInstance(project_state.models["test_alflpkfk", "pony"].get_field_by_name("id"), models.AutoField) self.assertIsInstance(new_state.models["test_alflpkfk", "pony"].get_field_by_name("id"), models.FloatField) - def assertIdTypeEqualsFkType(self): + def assertIdTypeEqualsFkType(): with connection.cursor() as cursor: id_type = [c.type_code for c in connection.introspection.get_table_description(cursor, "test_alflpkfk_pony") if c.name == "id"][0] fk_type = [c.type_code for c in connection.introspection.get_table_description(cursor, "test_alflpkfk_rider") if c.name == "pony_id"][0] self.assertEqual(id_type, fk_type) + assertIdTypeEqualsFkType() # Test the database alteration with connection.schema_editor() as editor: