diff options
Diffstat (limited to 'tests/postgres_tests')
| -rw-r--r-- | tests/postgres_tests/test_aggregates.py | 10 |
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 |
