def test_pluck(self): queryset = QuerySet(Item) expected = range(1, 31) self.assertItemsEqual(queryset.pluck('id'), expected) expected = range(10) * 3 self.assertItemsEqual(queryset.pluck('number'), expected)
def test_pluck_with_multiple_fields(self): queryset = QuerySet(Item).order_by('id')[:3] self.assertItemsEqual(queryset.pluck('id', 'number'), [(1, 1), (2, 2), (3, 3)])
def test_combined_filter_order_slice(self): queryset = QuerySet(Item).filter(number=2).order_by('id')[:2] self.assertEquals(list(queryset.pluck('id')), [2, 12])