diff options
Diffstat (limited to 'tests/select_related')
-rw-r--r-- | tests/select_related/tests.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/tests/select_related/tests.py b/tests/select_related/tests.py index 59b19d97f6..52aff92a7c 100644 --- a/tests/select_related/tests.py +++ b/tests/select_related/tests.py @@ -100,8 +100,9 @@ class SelectRelatedTests(TestCase): ['Agaricales', 'Diptera', 'Fabales', 'Primates']) def test_select_related_with_extra(self): - s = Species.objects.all().select_related()\ - .extra(select={'a': 'select_related_species.id + 10'})[0] + s = (Species.objects.all() + .select_related() + .extra(select={'a': 'select_related_species.id + 10'})[0]) self.assertEqual(s.id + 10, s.a) def test_certain_fields(self): @@ -131,12 +132,14 @@ class SelectRelatedTests(TestCase): def test_field_traversal(self): with self.assertNumQueries(1): - s = Species.objects.all().select_related('genus__family__order' - ).order_by('id')[0:1].get().genus.family.order.name + s = (Species.objects.all() + .select_related('genus__family__order') + .order_by('id')[0:1].get().genus.family.order.name) self.assertEqual(s, 'Diptera') def test_depth_fields_fails(self): - self.assertRaises(TypeError, + self.assertRaises( + TypeError, Species.objects.select_related, 'genus__family__order', depth=4 ) |