numpy.testing.assert_allclose#
- testing.assert_allclose(actual, desired, rtol=1e-07, atol=0, equal_nan=True, err_msg='', verbose=True, *, strict=False)[source]#
如果两个对象在所需的容差范围内不相同,则引发 AssertionError。
给定两个类似数组的对象,检查它们的形状和所有元素是否相等(但请参阅注释以了解标量的特殊处理)。如果形状不匹配或任何值冲突,则会引发异常。与 NumPy 中的标准用法相反,NaN 的比较方式与数字相同,如果两个对象在相同位置都有 NaN,则不会引发断言。
该测试等效于
allclose(actual, desired, rtol, atol)
(注意allclose
具有不同的默认值)。它将 actual 和 desired 之间的差值与atol + rtol * abs(desired)
进行比较。版本 1.5.0 中的新功能。
- 参数:
- actualarray_like
获得的数组。
- desiredarray_like
期望的数组。
- rtolfloat,可选
相对容差。
- atolfloat,可选
绝对容差。
- equal_nanbool,可选。
如果为 True,则 NaN 将被视为相等。
- err_msgstr,可选
失败时要打印的错误消息。
- verbosebool,可选
如果为 True,则冲突值将附加到错误消息中。
- strictbool,可选
如果为 True,则当参数的形状或数据类型不匹配时,引发
AssertionError
。注释部分中提到的标量的特殊处理将被禁用。版本 2.0.0 中的新功能。
- 引发:
- AssertionError
如果 actual 和 desired 在指定的精度范围内不相等。
注释
当 actual 和 desired 之一为标量,另一个为类似数组时,该函数执行比较,就像标量被广播到数组的形状一样。可以通过 strict 参数禁用此行为。
示例
>>> x = [1e-5, 1e-3, 1e-1] >>> y = np.arccos(np.cos(x)) >>> np.testing.assert_allclose(x, y, rtol=1e-5, atol=0)
如注释部分所述,
assert_allclose
对标量有特殊处理。这里,测试检查numpy.sin
的值在 π 的整数倍数处是否接近零。>>> x = np.arange(3) * np.pi >>> np.testing.assert_allclose(np.sin(x), 0, atol=1e-15)
使用 strict 对具有一个或多个维度的数组与标量进行比较时引发
AssertionError
。>>> np.testing.assert_allclose(np.sin(x), 0, atol=1e-15, strict=True) Traceback (most recent call last): ... AssertionError: Not equal to tolerance rtol=1e-07, atol=1e-15 (shapes (3,), () mismatch) ACTUAL: array([ 0.000000e+00, 1.224647e-16, -2.449294e-16]) DESIRED: array(0)
strict 参数还确保数组数据类型匹配
>>> y = np.zeros(3, dtype=np.float32) >>> np.testing.assert_allclose(np.sin(x), y, atol=1e-15, strict=True) Traceback (most recent call last): ... AssertionError: Not equal to tolerance rtol=1e-07, atol=1e-15 (dtypes float64, float32 mismatch) ACTUAL: array([ 0.000000e+00, 1.224647e-16, -2.449294e-16]) DESIRED: array([0., 0., 0.], dtype=float32)