Add test for return type for distinct GroupConcat

adamchainz · Jun 1, 2020 · ada23ea · ada23ea
1 parent 640da6f
commit ada23ea
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/testapp/test_aggregates.py b/tests/testapp/test_aggregates.py
@@ -76,6 +76,10 @@ def test_basic_aggregate_ids(self):
         out = self.shakes.tutees.aggregate(tids=GroupConcat("id"))
         assert out == {"tids": self.str_tutee_ids}
 
+    def test_distinct_aggregate_ids(self):
+        out = self.shakes.tutees.aggregate(tids=GroupConcat("id", distinct=True))
+        assert out == {"tids": set(self.str_tutee_ids)}
+
     def test_basic_annotate_ids(self):
         concat = GroupConcat("tutees__id")
         shakey = Author.objects.annotate(tids=concat).get(id=self.shakes.id)