diff --git a/tests/expressions_case/tests.py b/tests/expressions_case/tests.py
index 58dc7ca..8c081dd 100644
a
|
b
|
class CaseExpressionTests(TestCase):
|
240 | 240 | transform=itemgetter('integer', 'max', 'test') |
241 | 241 | ) |
242 | 242 | |
| 243 | def test_annotate_exclude(self): |
| 244 | self.assertQuerysetEqual( |
| 245 | CaseTestModel.objects.annotate(test=Case( |
| 246 | When(integer=1, then=Value('one')), |
| 247 | When(integer=2, then=Value('two')), |
| 248 | default=Value('other'), |
| 249 | output_field=models.CharField(), |
| 250 | )).exclude(test='other').order_by('pk'), |
| 251 | [(1, 'one'), (2, 'two'), (2, 'two')], |
| 252 | transform=attrgetter('integer', 'test') |
| 253 | ) |
| 254 | |
243 | 255 | def test_combined_expression(self): |
244 | 256 | self.assertQuerysetEqual( |
245 | 257 | CaseTestModel.objects.annotate( |