summaryrefslogtreecommitdiff
path: root/tests/postgres_tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests/postgres_tests')
-rw-r--r--tests/postgres_tests/test_aggregates.py10
1 files changed, 9 insertions, 1 deletions
diff --git a/tests/postgres_tests/test_aggregates.py b/tests/postgres_tests/test_aggregates.py
index 9aa0e06595..056d08441b 100644
--- a/tests/postgres_tests/test_aggregates.py
+++ b/tests/postgres_tests/test_aggregates.py
@@ -128,7 +128,7 @@ class TestGeneralAggregate(PostgreSQLTestCase):
self.assertEqual(values, json.loads('{"jsonagg": []}'))
-class TestStringAggregateDistinct(PostgreSQLTestCase):
+class TestAggregateDistinct(PostgreSQLTestCase):
@classmethod
def setUpTestData(cls):
AggregateTestModel.objects.create(char_field='Foo')
@@ -145,6 +145,14 @@ class TestStringAggregateDistinct(PostgreSQLTestCase):
self.assertEqual(values['stringagg'].count('Foo'), 1)
self.assertEqual(values['stringagg'].count('Bar'), 1)
+ def test_array_agg_distinct_false(self):
+ values = AggregateTestModel.objects.aggregate(arrayagg=ArrayAgg('char_field', distinct=False))
+ self.assertEqual(sorted(values['arrayagg']), ['Bar', 'Foo', 'Foo'])
+
+ def test_array_agg_distinct_true(self):
+ values = AggregateTestModel.objects.aggregate(arrayagg=ArrayAgg('char_field', distinct=True))
+ self.assertEqual(sorted(values['arrayagg']), ['Bar', 'Foo'])
+
class TestStatisticsAggregate(PostgreSQLTestCase):
@classmethod