diff options
Diffstat (limited to 'tests/m2m_recursive/tests.py')
| -rw-r--r-- | tests/m2m_recursive/tests.py | 46 |
1 files changed, 30 insertions, 16 deletions
diff --git a/tests/m2m_recursive/tests.py b/tests/m2m_recursive/tests.py index d0799f07d2..d5ebb500c1 100644 --- a/tests/m2m_recursive/tests.py +++ b/tests/m2m_recursive/tests.py @@ -10,7 +10,7 @@ class RecursiveM2MTests(TestCase): def setUpTestData(cls): cls.a, cls.b, cls.c, cls.d = [ Person.objects.create(name=name) - for name in ['Anne', 'Bill', 'Chuck', 'David'] + for name in ["Anne", "Bill", "Chuck", "David"] ] cls.a.friends.add(cls.b, cls.c) # Add m2m for Anne and Chuck in reverse direction. @@ -68,15 +68,23 @@ class RecursiveSymmetricalM2MThroughTests(TestCase): def setUpTestData(cls): cls.a, cls.b, cls.c, cls.d = [ Person.objects.create(name=name) - for name in ['Anne', 'Bill', 'Chuck', 'David'] + for name in ["Anne", "Bill", "Chuck", "David"] ] - cls.a.colleagues.add(cls.b, cls.c, through_defaults={ - 'first_meet': datetime.date(2013, 1, 5), - }) + cls.a.colleagues.add( + cls.b, + cls.c, + through_defaults={ + "first_meet": datetime.date(2013, 1, 5), + }, + ) # Add m2m for Anne and Chuck in reverse direction. - cls.d.colleagues.add(cls.a, cls.c, through_defaults={ - 'first_meet': datetime.date(2015, 6, 15), - }) + cls.d.colleagues.add( + cls.a, + cls.c, + through_defaults={ + "first_meet": datetime.date(2015, 6, 15), + }, + ) def test_recursive_m2m_all(self): for person, colleagues in ( @@ -90,9 +98,12 @@ class RecursiveSymmetricalM2MThroughTests(TestCase): def test_recursive_m2m_reverse_add(self): # Add m2m for Anne in reverse direction. - self.b.colleagues.add(self.a, through_defaults={ - 'first_meet': datetime.date(2013, 1, 5), - }) + self.b.colleagues.add( + self.a, + through_defaults={ + "first_meet": datetime.date(2013, 1, 5), + }, + ) self.assertSequenceEqual(self.a.colleagues.all(), [self.b, self.c, self.d]) self.assertSequenceEqual(self.b.colleagues.all(), [self.a]) @@ -111,9 +122,12 @@ class RecursiveSymmetricalM2MThroughTests(TestCase): def test_recursive_m2m_set(self): # Set new relationships for Chuck. - self.c.colleagues.set([self.b, self.d], through_defaults={ - 'first_meet': datetime.date(2013, 1, 5), - }) - self.assertSequenceEqual(self.c.colleagues.order_by('name'), [self.b, self.d]) + self.c.colleagues.set( + [self.b, self.d], + through_defaults={ + "first_meet": datetime.date(2013, 1, 5), + }, + ) + self.assertSequenceEqual(self.c.colleagues.order_by("name"), [self.b, self.d]) # Reverse m2m relationships is removed. - self.assertSequenceEqual(self.a.colleagues.order_by('name'), [self.b, self.d]) + self.assertSequenceEqual(self.a.colleagues.order_by("name"), [self.b, self.d]) |