diff options
Diffstat (limited to 'tests/annotations')
| -rw-r--r-- | tests/annotations/tests.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/annotations/tests.py b/tests/annotations/tests.py index 65886b2063..b149ddf46c 100644 --- a/tests/annotations/tests.py +++ b/tests/annotations/tests.py @@ -6,7 +6,7 @@ from django.db.models import ( BooleanField, CharField, Count, DateTimeField, ExpressionWrapper, F, Func, IntegerField, NullBooleanField, Q, Sum, Value, ) -from django.db.models.functions import Lower +from django.db.models.functions import Length, Lower from django.test import TestCase, skipUnlessDBFeature from .models import ( @@ -205,6 +205,11 @@ class NonAggregateAnnotationTestCase(TestCase): ).distinct('test_alias') self.assertEqual(len(people2), 1) + lengths = Employee.objects.annotate( + name_len=Length('first_name'), + ).distinct('name_len').values_list('name_len', flat=True) + self.assertSequenceEqual(lengths, [3, 7, 8]) + def test_filter_annotation(self): books = Book.objects.annotate( is_book=Value(1, output_field=IntegerField()) |