1. from django.db.models import Q

  2. from django.test import TestCase

  3. 

  4. from .models import Number

  5. 

  6. 

  7. class XorLookupsTests(TestCase):

  8.     @classmethod

  9.     def setUpTestData(cls):

  10.         cls.numbers = [Number.objects.create(num=i) for i in range(10)]

  11. 

  12.     def test_filter(self):

  13.         self.assertCountEqual(

  14.             Number.objects.filter(num__lte=7) ^ Number.objects.filter(num__gte=3),

  15.             self.numbers[:3] + self.numbers[8:],

  16.         )

  17.         self.assertCountEqual(

  18.             Number.objects.filter(Q(num__lte=7) ^ Q(num__gte=3)),

  19.             self.numbers[:3] + self.numbers[8:],

  20.         )

  21. 

  22.     def test_filter_negated(self):

  23.         self.assertCountEqual(

  24.             Number.objects.filter(Q(num__lte=7) ^ ~Q(num__lt=3)),

  25.             self.numbers[:3] + self.numbers[8:],

  26.         )

  27.         self.assertCountEqual(

  28.             Number.objects.filter(~Q(num__gt=7) ^ ~Q(num__lt=3)),

  29.             self.numbers[:3] + self.numbers[8:],

  30.         )

  31.         self.assertCountEqual(

  32.             Number.objects.filter(Q(num__lte=7) ^ ~Q(num__lt=3) ^ Q(num__lte=1)),

  33.             [self.numbers[2]] + self.numbers[8:],

  34.         )

  35.         self.assertCountEqual(

  36.             Number.objects.filter(~(Q(num__lte=7) ^ ~Q(num__lt=3) ^ Q(num__lte=1))),

  37.             self.numbers[:2] + self.numbers[3:8],

  38.         )

  39. 

  40.     def test_exclude(self):

  41.         self.assertCountEqual(

  42.             Number.objects.exclude(Q(num__lte=7) ^ Q(num__gte=3)),

  43.             self.numbers[3:8],

  44.         )

  45. 

  46.     def test_stages(self):

  47.         numbers = Number.objects.all()

  48.         self.assertSequenceEqual(

  49.             numbers.filter(num__gte=0) ^ numbers.filter(num__lte=11),

  50.             [],

  51.         )

  52.         self.assertSequenceEqual(

  53.             numbers.filter(num__gt=0) ^ numbers.filter(num__lt=11),

  54.             [self.numbers[0]],

  55.         )

  56. 

  57.     def test_pk_q(self):

  58.         self.assertCountEqual(

  59.             Number.objects.filter(Q(pk=self.numbers[0].pk) ^ Q(pk=self.numbers[1].pk)),

  60.             self.numbers[:2],

  61.         )

  62. 

  63.     def test_empty_in(self):

  64.         self.assertCountEqual(

  65.             Number.objects.filter(Q(pk__in=[]) ^ Q(num__gte=5)),

  66.             self.numbers[5:],

  67.         )
About Sign Up Leave Feedback Contact