对于多条件断言,我们可以使用assert_allclose()函数来代替assert语句。例如:
import numpy as np a = np.array([1., 2., 3., 4.]) b = np.array([1.01, 1.99, 3.03, 4.]) assert np.allclose(a, b, rtol=0.05)
这里的assert_allclose()函数用于确定两个数组是否在精度范围内相等。rtol参数设置相对精度,假设这里为0.05,表示两个数字相差在5%以内就视为相等。如果上述条件不满足,则会引发一个AssertionError。因此,不再需要使用多个assert语句来检查多个条件。
上一篇:assert行为的问题