如何测试预期抛出IndexError的类方法，现有断言写法不生效

你原来的写法存在两个核心问题：

1. 直接调用my_class.calculate_score(input_data)时，如果方法真的抛出IndexError，代码会直接终止运行，不会执行后续的断言逻辑
2. assert(IndexError, result)的用法完全不符合预期，这个语句等价于判断bool(IndexError)，而这个值永远为True，根本不会校验是否抛出了目标异常

以下是三种常用的正确实现方式：

方案1：使用Python内置unittest框架 #

import unittest

class TestStudentGrades(unittest.TestCase):
    def test_calculate_score_empty_input_raise_index_error(self):
        input_data = []
        my_class = StudentGrades()
        # 用assertRaises上下文管理器捕获预期异常
        with self.assertRaises(IndexError):
            my_class.calculate_score(input_data)

if __name__ == '__main__':
    unittest.main()

如果执行方法时抛出了IndexError则测试通过，没抛异常或者抛了其他类型的异常则测试失败。

方案2：使用pytest框架 #

import pytest

def test_calculate_score_empty_input_raise_index_error():
    input_data = []
    my_class = StudentGrades()
    with pytest.raises(IndexError):
        my_class.calculate_score(input_data)

运行pytest命令即可执行测试，判断逻辑和unittest完全一致。

方案3：不依赖测试框架，手动捕获异常验证 #

如果你不想引入测试框架，也可以用try-except手动实现校验逻辑：

input_data = []
my_class = StudentGrades()
has_correct_error = False
try:
    my_class.calculate_score(input_data)
except IndexError:
    has_correct_error = True
except Exception as e:
    # 抛出了其他类型的异常，不符合预期
    raise AssertionError(f"预期抛出IndexError，实际抛出了{type(e).__name__}") from e

assert has_correct_error, "预期抛出IndexError，但实际没有抛出任何异常"

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