Django snippets: assertQuerysetEqual

from django.test import TestCase

class BaseTestCase(TestCase):
    def assertQuerysetEqual(self, qs1, qs2):
        pk = lambda o: o.pk
        return self.assertEqual(
            list(sorted(qs1, key=pk)),
            list(sorted(qs2, key=pk))
        )

More like this

Fuzzy testing with assertNumQueries by spookylukey 1 year ago
Decorator and context manager to override settings by jezdez 8 months, 4 weeks ago
"Zoom in" on rendered HTML that the test client returns by peterbe 2 years, 9 months ago
Multiple querysets by t_rybik 1 year, 11 months ago
FieldLevelPermissionsAdmin by buriy 4 years, 5 months ago

I often find myself testing that the queryset returned by a method contains the instances I expect. I use a custom method, assertQuerysetEqual(), to test the equality of two querysets or lists::

def test_some_values(self):
    qs = get_user_list()
    self.assertQuerysetEqual(qs, [normal_user, super_user])

Makes it easy to test small querysets against lists whose values are known and expected.

Author:: coleifer
Posted:: May 7, 2010
Language:: Python
Django Version:: 1.1
Tags:: test testing
Score:: 3 (after 3 ratings)

Django snippets

assertQuerysetEqual

More like this

Tools

Comments

Snippets

About