10

有没有办法在 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

为什么其他阵列的测试失败？

4

2 回答 2

13

测试两个列表

使用：assertSequenceEqual

因为，在这种情况下，tags = Tag.objects.all()生成一个django.db.models.query.QuerySetwhere astag_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

于 2014-06-04T04:12:38.927 回答
3

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

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

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

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

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

于 2014-06-02T04:29:17.673 回答