From 910638fc4e4f83bda31dd91db5017acb06dfc8d2 Mon Sep 17 00:00:00 2001 From: Edward Henderson Date: Tue, 14 Apr 2015 08:51:48 -0600 Subject: [PATCH] Refs #24385 -- Added tests for distinct sum issue fixed in c7fd9b242d2d63406f1de6cc3204e35aaa025233 --- tests/aggregation/tests.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/aggregation/tests.py b/tests/aggregation/tests.py index 424b7a7ccc6..3e2bc2eb309 100644 --- a/tests/aggregation/tests.py +++ b/tests/aggregation/tests.py @@ -441,6 +441,20 @@ class AggregateTestCase(TestCase): vals = Book.objects.annotate(num_authors=Count("authors__id")).aggregate(Avg("num_authors")) self.assertEqual(vals, {"num_authors__avg": Approximate(1.66, places=1)}) + def test_sum_distinct_aggregate(self): + """ + Sum on a distict() QuerySet should aggregate only the distinct items. + """ + authors = Author.objects.filter(book__in=[5, 6]) + self.assertEqual(authors.count(), 3) + + distinct_authors = authors.distinct() + self.assertEqual(distinct_authors.count(), 2) + + # Selected author ages are 57 and 46 + age_sum = distinct_authors.aggregate(Sum('age')) + self.assertEqual(age_sum['age__sum'], 103) + def test_filtering(self): p = Publisher.objects.create(name='Expensive Publisher', num_awards=0) Book.objects.create(