Python unittest自定义异常assertRaises断言失败问题求助

嘿，这种情况我太熟了——明明代码里确实抛出了自定义的MyException，但测试就是不认，换成Exception就过了，大概率是几个容易踩的小坑在搞鬼。咱们一步步来排查：

1. 最常见的坑：导入路径不一致导致类不匹配 #

这是90%概率会碰到的问题：你的测试代码里引用的MyException，和foo.py里抛出的MyException根本不是同一个类对象！

比如你的项目结构是：

my_project/
├── exceptions.py
├── foo.py
└── test_foo.py

如果foo.py里用的是相对导入：

from .exceptions import MyException

而test_foo.py里用的是绝对导入（或者反过来）：

from exceptions import MyException

在某些测试环境下（比如直接运行python test_foo.py而不是用pytest之类的框架处理路径），这会导致两个模块加载的exceptions.py不是同一个，进而生成两个不同的MyException类——测试断言的时候自然会失败，但因为两个类都是Exception的子类，换成Exception就能通过。

快速验证方法：
在foo.py的抛出异常处添加打印：

# foo.py 中抛出异常的代码段
exc = MyException("测试异常")
print(f"抛出的异常类: {exc.__class__}, 模块: {exc.__class__.__module__}, ID: {id(exc.__class__)}")
raise exc

在test_foo.py的测试用例前添加打印：

# test_foo.py
from exceptions import MyException
print(f"测试用的异常类: {MyException}, 模块: {MyException.__module__}, ID: {id(MyException)}")

def test_something():
    with pytest.raises(MyException):
        foo.some_function()

运行测试后对比两个ID值，如果不一样，就实锤是导入路径的问题了。

解决办法：
统一项目中的导入方式，比如都用绝对导入（假设你的项目根目录在Python路径中）：

# foo.py 和 test_foo.py 都用这个导入
from my_project.exceptions import MyException

或者确保测试运行时的工作目录正确，让相对导入能指向同一个模块。

2. 不小心重复定义了异常类 #

检查一下foo.py里有没有不小心重新定义了MyException——比如你可能在foo.py里又写了一遍：

# foo.py 里的错误代码
class MyException(Exception):
    pass

def some_function():
    raise MyException()

这时候foo.py里抛出的是本地定义的MyException，而测试代码里导入的是exceptions.py中的类，两者完全不是一回事，断言自然失败。

3. 测试框架的断言用法有误 #

不同测试框架的断言方式略有不同，确认你用对了：

• unittest框架：必须用上下文管理器或者传递可调用对象：

  # 正确用法
  def test_my_exception(self):
      with self.assertRaises(MyException):
          foo.some_function()
  

• pytest框架：同样推荐用上下文管理器：

  def test_my_exception():
      with pytest.raises(MyException):
          foo.some_function()
  

要是你错误地把异常实例当成参数传进去（比如assertRaises(MyException())），那肯定会失败，但换成Exception的时候可能碰巧蒙对？不过这种情况比较少见。

4. 小概率：模块重复加载问题 #

如果你的项目结构比较复杂，或者用了一些自定义的导入逻辑，可能导致exceptions.py被加载了两次，生成了两个独立的MyException类。这种情况可以通过检查sys.modules来验证：
在foo.py和test_foo.py里分别打印：

import sys
print(sys.modules.get('exceptions'))

如果输出的模块对象不一样，说明模块被重复加载了，这时候需要调整导入逻辑或者项目结构。

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