python - 断言 numpy.array 相等性的最佳方法？

Question

我想为我的应用做一些单元测试，我需要比较两个数组。由于 array.__eq__ 返回一个新数组(所以 TestCase.assertEqual 失败)，断言相等的最佳方法是什么？

目前我正在使用

self.assertTrue((arr1 == arr2).all())

但我不是很喜欢它

Answer 1

查看 numpy.testing 中的断言函数，例如

assert_array_equal

对于 float 组相等性测试可能会失败，而 assert_almost_equal 更可靠。

更新

在几个版本之前，numpy 获得了 assert_allclose，它现在是我最喜欢的，因为它允许我们指定绝对误差和相对误差，并且不需要小数舍入作为接近标准。