diff options
Diffstat (limited to 'tests/m2o_recursive/tests.py')
| -rw-r--r-- | tests/m2o_recursive/tests.py | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/tests/m2o_recursive/tests.py b/tests/m2o_recursive/tests.py index 1b9a72a796..ff68c60ee6 100644 --- a/tests/m2o_recursive/tests.py +++ b/tests/m2o_recursive/tests.py @@ -4,27 +4,31 @@ from .models import Category, Person class ManyToOneRecursiveTests(TestCase): - @classmethod def setUpTestData(cls): - cls.r = Category.objects.create(id=None, name='Root category', parent=None) - cls.c = Category.objects.create(id=None, name='Child category', parent=cls.r) + cls.r = Category.objects.create(id=None, name="Root category", parent=None) + cls.c = Category.objects.create(id=None, name="Child category", parent=cls.r) def test_m2o_recursive(self): self.assertSequenceEqual(self.r.child_set.all(), [self.c]) - self.assertEqual(self.r.child_set.get(name__startswith='Child').id, self.c.id) + self.assertEqual(self.r.child_set.get(name__startswith="Child").id, self.c.id) self.assertIsNone(self.r.parent) self.assertSequenceEqual(self.c.child_set.all(), []) self.assertEqual(self.c.parent.id, self.r.id) class MultipleManyToOneRecursiveTests(TestCase): - @classmethod def setUpTestData(cls): - cls.dad = Person.objects.create(full_name='John Smith Senior', mother=None, father=None) - cls.mom = Person.objects.create(full_name='Jane Smith', mother=None, father=None) - cls.kid = Person.objects.create(full_name='John Smith Junior', mother=cls.mom, father=cls.dad) + cls.dad = Person.objects.create( + full_name="John Smith Senior", mother=None, father=None + ) + cls.mom = Person.objects.create( + full_name="Jane Smith", mother=None, father=None + ) + cls.kid = Person.objects.create( + full_name="John Smith Junior", mother=cls.mom, father=cls.dad + ) def test_m2o_recursive2(self): self.assertEqual(self.kid.mother.id, self.mom.id) |
