使用unittest测试空numpy数组时触发AssertionError的解决方法咨询

这个问题我之前也碰到过！问题出在unittest.assertEqual并不适合直接比较numpy数组——哪怕两个空数组看起来完全一致，它也没法正确识别numpy数组的内部结构，所以才会出现看起来一样却断言失败的情况。

正确的解决方案：使用numpy专属的断言方法 #

numpy专门提供了用于数组比较的测试工具，最常用的是np.testing.assert_array_equal，它会完整检查两个数组的形状、数据类型、元素值（空数组的话就验证形状和dtype匹配），完美适配numpy的特性。

修改你的测试代码如下：

import unittest
import numpy as np

def myFunction(var):
    # 示例函数实现：var=0返回空int64数组，其他值返回非空数组
    if var == 0:
        return np.array([], dtype=np.int64)
    else:
        return np.array([1, 2, 3], dtype=np.int64)

class myTests(unittest.TestCase):
    def test_empty_array_case(self):
        var = 0
        result = myFunction(var)
        # 使用numpy的断言方法替代unittest.assertEqual
        np.testing.assert_array_equal(np.array([], dtype=np.int64), result)
    
    def test_non_empty_array_case(self):
        var = 1
        expected = np.array([1, 2, 3], dtype=np.int64)
        result = myFunction(var)
        np.testing.assert_array_equal(expected, result)

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

另一种可选方案：结合np.array_equal和self.assertTrue #

如果你更习惯用unittest自带的断言风格，也可以用np.array_equal函数先判断数组是否一致，再用self.assertTrue验证结果：

def test_empty_array_case(self):
    var = 0
    result = myFunction(var)
    self.assertTrue(np.array_equal(np.array([], dtype=np.int64), result))

这个方法的原理是np.array_equal会返回一个布尔值（而不是numpy布尔数组），告诉我们两个数组是否完全匹配，刚好符合self.assertTrue的要求。

避坑提醒 #

别直接用==比较numpy数组后传给self.assertTrue！比如self.assertTrue(np.array([], dtype=np.int64) == result)会报错，因为==对空数组会返回一个空的布尔数组，而不是单个True/False，unittest无法识别这种类型的断言输入。

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