Python Unittest的assertEqual和可迭代类型 – 仅检查内容

在本文中，我们将介绍Python的unittest模块中的assertEqual方法以及如何使用它来测试可迭代类型的内容。

assertEqual方法简介

unittest是Python中一个流行的单元测试框架。它提供了许多用于测试代码的方法和断言。其中之一是assertEqual方法，用于断言两个值是否相等。

assertEqual方法的语法如下：

self.assertEqual(a, b)

其中，a是我们期望的值，b是实际得到的值。如果a和b相等，则测试通过，否则测试失败。

使用assertEqual测试可迭代类型的内容

除了基本的数据类型，assertEqual方法也可以用于测试可迭代类型（如列表、元组和集合）。

让我们通过一个例子来演示如何使用assertEqual方法测试列表的内容是否相等：

import unittest

class MyTestCase(unittest.TestCase):
    def test_list_equal(self):
        expected = [1, 2, 3]
        actual = [1, 2, 3]
        self.assertEqual(expected, actual)

上述代码定义了一个名为MyTestCase的测试用例类，并在其中定义了一个test_list_equal方法。在这个方法中，我们定义了期望的列表expected和实际得到的列表actual。通过调用assertEqual方法，我们断言这两个列表的内容是否相等。

除了列表，使用assertEqual方法测试元组和集合的内容也是类似的。

仅检查内容

assertEqual方法会对可迭代类型的内容进行完全比较，包括元素的顺序和值是否完全一致。但是，在某些情况下，我们可能只关心可迭代类型的内容是否相等，而不关心顺序。

unittest模块中提供了其他一些方法来实现这个功能。其中一个是assertCountEqual方法，它只关注可迭代类型的内容是否相等，而不关心顺序。让我们通过一个例子来说明：

import unittest

class MyTestCase(unittest.TestCase):
    def test_set_equal(self):
        expected = {1, 2, 3}
        actual = {3, 2, 1}
        self.assertCountEqual(expected, actual)