Refs #28930 -- Simplified schemas test with any().

This commit is contained in:
Tim Graham 2018-01-02 10:11:48 -05:00
parent f5a989e603
commit 5f456408a0
1 changed files with 2 additions and 8 deletions

View File

@ -1469,10 +1469,7 @@ class SchemaTests(TransactionTestCase):
editor.create_model(Author) editor.create_model(Author)
# Ensure the constraint exists # Ensure the constraint exists
constraints = self.get_constraints(Author._meta.db_table) constraints = self.get_constraints(Author._meta.db_table)
for details in constraints.values(): if not any(details['columns'] == ['height'] and details['check'] for details in constraints.values()):
if details['columns'] == ["height"] and details['check']:
break
else:
self.fail("No check constraint for height found") self.fail("No check constraint for height found")
# Alter the column to remove it # Alter the column to remove it
old_field = Author._meta.get_field("height") old_field = Author._meta.get_field("height")
@ -1489,10 +1486,7 @@ class SchemaTests(TransactionTestCase):
with connection.schema_editor() as editor: with connection.schema_editor() as editor:
editor.alter_field(Author, new_field, new_field2, strict=True) editor.alter_field(Author, new_field, new_field2, strict=True)
constraints = self.get_constraints(Author._meta.db_table) constraints = self.get_constraints(Author._meta.db_table)
for details in constraints.values(): if not any(details['columns'] == ['height'] and details['check'] for details in constraints.values()):
if details['columns'] == ["height"] and details['check']:
break
else:
self.fail("No check constraint for height found") self.fail("No check constraint for height found")
def test_unique(self): def test_unique(self):