diff options
Diffstat (limited to 'tests/postgres_tests')
| -rw-r--r-- | tests/postgres_tests/test_aggregates.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/postgres_tests/test_aggregates.py b/tests/postgres_tests/test_aggregates.py index 056d08441b..d4a01ff027 100644 --- a/tests/postgres_tests/test_aggregates.py +++ b/tests/postgres_tests/test_aggregates.py @@ -46,6 +46,18 @@ class TestGeneralAggregate(PostgreSQLTestCase): values = AggregateTestModel.objects.aggregate(arrayagg=ArrayAgg('boolean_field')) self.assertEqual(values, {'arrayagg': []}) + def test_array_agg_lookups(self): + aggr1 = AggregateTestModel.objects.create() + aggr2 = AggregateTestModel.objects.create() + StatTestModel.objects.create(related_field=aggr1, int1=1, int2=0) + StatTestModel.objects.create(related_field=aggr1, int1=2, int2=0) + StatTestModel.objects.create(related_field=aggr2, int1=3, int2=0) + StatTestModel.objects.create(related_field=aggr2, int1=4, int2=0) + qs = StatTestModel.objects.values('related_field').annotate( + array=ArrayAgg('int1') + ).filter(array__overlap=[2]).values_list('array', flat=True) + self.assertCountEqual(qs.get(), [1, 2]) + def test_bit_and_general(self): values = AggregateTestModel.objects.filter( integer_field__in=[0, 1]).aggregate(bitand=BitAnd('integer_field')) |