Python数组列表append方法单元测试：如何验证断言错误？

嗨，作为刚上手Python类和单元测试的新手，你的问题很有代表性，我来帮你梳理清楚：

一、如何检查append方法的断言错误？ #

除了连续追加50次触发断言外，更高效的方式是利用测试框架的异常捕获机制，直接模拟列表已满的状态，然后验证append是否抛出AssertionError。主流的Python测试框架（unittest或pytest）都支持这种操作：

示例代码（以unittest为例） #

假设你的列表类是MyArrayList，append方法完整代码大概是这样：

class MyArrayList:
    def __init__(self):
        self.capacity = 50
        self.array = [None] * self.capacity
        self.count = 0  # 下一个可用位置

    def append(self, item):
        assert self.count <= self.capacity, "列表容量已达上限"
        self.array[self.count] = item
        self.count += 1

对应的单元测试可以这么写：

import unittest

class TestMyArrayList(unittest.TestCase):
    def setUp(self):
        # 每次测试前创建一个新的列表实例
        self.my_list = MyArrayList()

    def test_append_when_list_is_full(self):
        # 手动将count设置为容量上限，模拟列表已满的状态
        self.my_list.count = 50
        
        # 使用上下文管理器捕获AssertionError
        with self.assertRaises(AssertionError) as error_context:
            self.my_list.append("测试元素")
        
        # 可选：验证断言的错误消息是否符合预期
        self.assertEqual(str(error_context.exception), "列表容量已达上限")

    def test_append_51st_item(self):
        # 你提到的连续追加50次的方式，作为补充测试
        for i in range(50):
            self.my_list.append(i)
        
        with self.assertRaises(AssertionError):
            self.my_list.append("第51个元素")

如果用pytest，代码会更简洁：

import pytest

def test_append_full_list():
    my_list = MyArrayList()
    my_list.count = 50
    with pytest.raises(AssertionError, match="列表容量已达上限"):
        my_list.append("测试元素")

二、是否可以手动修改self.count？ #

当然可以！在单元测试中，手动修改实例的属性是非常常见的测试技巧——我们的目标是覆盖边界场景，而不是必须通过正常业务逻辑来触发状态。直接把self.count设为50，能快速模拟列表已满的状态，避免循环50次的冗余操作，让测试更高效。

不过要注意：这种操作只适合在测试代码中使用，不要在业务逻辑里手动修改self.count，否则会破坏列表的内部一致性。

内容的提问来源于stack exchange，提问作者Sook Lim