如何使用pytest-django测试Django Queryset相等

Question

断言两个查询集相等的最佳/最易读方式是什么？我想出了几个解决方案：

# option 1
assert sorted(qs1.values_list("pk", flat=True)) == sorted(qs2.values_list("pk", flat=True))

# option 2 (need to assert length first because set might remove duplicates)
assert len(qs1) == len(qs2)
assert set(qs1) == set(qs2)

我知道Django有一个方法django.test.TransactionTestCase.assertQuerysetEqual。pytest-django也有类似的东西吗？我在文档里没有看到它。

Answer 1

就在你建议的链接的起跑线上：

断言 Django的所有TestCase断言都可以在pytest_django.asserts中使用，例如from pytest_django.asserts import assertTemplateUsed

类似地，您可以从pytest_django.asserts import assertQuerysetEqual中使用