Commit 7515f657 authored by Alex Gaynor's avatar Alex Gaynor
Browse files

Fix TestCase.assertQuerysetEqual on python 3, this is needed for a large number of tests

parent 4c97101b
Loading
Loading
Loading
Loading
+3 −2
Original line number Diff line number Diff line
@@ -796,9 +796,10 @@ class TransactionTestCase(SimpleTestCase):
            " the response" % template_name)

    def assertQuerysetEqual(self, qs, values, transform=repr, ordered=True):
        items = six.moves.map(transform, qs)
        if not ordered:
            return self.assertEqual(set(map(transform, qs)), set(values))
        return self.assertEqual(map(transform, qs), values)
            return self.assertEqual(set(items), set(values))
        return self.assertEqual(list(items), values)

    def assertNumQueries(self, num, func=None, *args, **kwargs):
        using = kwargs.pop("using", DEFAULT_DB_ALIAS)