断言两个对象列表在 Django 测试中是相等的

Question

有没有办法在 django 测试中检查两个对象列表是否相等。

假设我有一些模型：

class Tag(models.Model):
    slug = models.SlugField(max_length=50, unique=True)
    def __unicode__(self):
        return self.slug

然后我运行这个简单的测试：

def test_equal_list_fail(self):
    tag_list = []
    for tag in ['a', 'b', 'c']:
        tag_list.append(Tag.objects.create(slug=tag))

    tags = Tag.objects.all()

    self.assertEqual(tag_list, tags)

这失败了：

======================================================================
FAIL: test_equal_list_fail (userAccount.tests.ProfileTests)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "path/to/tests.py", line 155, in test_equal_list_fail
    self.assertEqual(tag_list, tags)
AssertionError: [<Tag: a>, <Tag: b>, <Tag: c>] != [<Tag: a>, <Tag: b>, <Tag: c>]

----------------------------------------------------------------------

这将起作用：

def test_equal_list_passes(self):
    tag_list = []
    for tag in ['a', 'b', 'c']:
        tag_list.append(Tag.objects.create(slug=tag))

    tags = Tag.objects.all()

    for tag_set in zip(tags, tag_list):
        self.assertEqual(*tag_set)

但是，这失败了：

def test_equal_list_fail(self):
    tag_list = []
    for tag in ['a', 'b', 'c']:
        tag_list.append(Tag.objects.create(slug=tag))

    tags = Tag.objects.all()

    for tag_set in zip(tags, tag_list):
        print "\n"
        print tag_set[0].slug + "'s pk is %s" % tag_set[0].pk
        print tag_set[1].slug + "'s pk is %s" % tag_set[1].pk
        print "\n"
        self.assertIs(*tag_set)

与：

Creating test database for alias 'default'...
.......

a's pk is 1
a's pk is 1

F.
======================================================================
FAIL: test_equal_list_fail (userAccount.tests.ProfileTests)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "path/to/tests.py", line 160, in test_equal_list_fail
    self.assertIs(*tag_set)
AssertionError: <Tag: a> is not <Tag: a>

这是预期的行为吗？

编辑回应评论
这种比较有效：

class Obj:
    def __init__(self, x):
        self.x = x

>>> one = Obj(1)
>>> two = Obj(2)
>>> a = [one, two]
>>> b = [one, two]
>>> a == b
True

为什么其他数组的测试失败了？

Answer 1

测试两个列表

使用：assertSequenceEqual

因为在这种情况下，tags = Tag.objects.all() 会生成一个 django.db.models.query.QuerySet，而 tag_list.append(...) 会创建一个列表。

不同情况下的其他选项有：

• assertListEqual(a, b)
• assertTupleEqual(a, b)
• assertSetEqual(a, b)
• assertDictEqual(a, b)

为什么<Tag: a> is not <Tag: a>

标签是相同的模型，但它们被加载到内存中的不同位置

for tag_set in zip(tags, tag_list):
    print "\n"
    print tag_set[0].slug + "'s pk is %s" % tag_set[0].pk + ' id is: ' + id(tag_set[0])
    print tag_set[1].slug + "'s pk is %s" % tag_set[1].pk + ' id is: ' + id(tag_set[1])
    print "\n"
    self.assertIs(*tag_set)

返回

.......

a's pk is 1 id is: 4522000208
a's pk is 1 id is: 4522228112

F.

因此，is 将返回 False

Answer 2

我认为您要测试的是创建的标签是否与您的测试列表中的标签具有相同的 slug。

为此，使用values_list 仅获取 slug 作为列表，然后进行比较：

assertEqual(Tag.objects.values_list('slug', flat=True), ['a','b','c'])

我不得不说，这不是一个很有用的测试，因为您正在检查 django orm 功能，该功能已经过很好的测试。

您的测试应该检查您自己的应用程序的细节。