Python 'assert' 函数 ValueError:具有多个元素的数组的真值是不明确的.使用 a.any() 或

Question

我正在尝试用Python编写一个assert函数来测试我的函数的输出是否是特定的值数组。

使用assert simulate(15,0,3) == np.array([15.,15.,15.,15.])，我得到以下错误：

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

如何测试函数输出是否为具有多个值的数组？

Answer 1

使用 numpy 时，assert函数效果不佳。请改用allclose()。

这应该有效

print(np.allclose(simulate(15,0,3), np.array([15,15,15,15])))