def test_padded_queryset_limiting(self): long_list = Permission.objects.all() self.assertTrue(long_list.count() > 6) padded_list = utils.padded_queryset(long_list) self.assertEqual(len(padded_list), 6)
def test_padded_queryset(self): short_list = get_user_model().objects.all()[:1] padded_list = utils.padded_queryset(short_list) expected_list = list(short_list) + [None, None, None, None, None] self.assertEqual(padded_list, expected_list)