提交 58be77e9 authored 作者: notoraptor's avatar notoraptor

Use `utt.assert_allclose()` instead of `assert np.all` in a test.

上级 375fec9a
......@@ -330,7 +330,7 @@ class Test_TopK(unittest.TestCase):
goal = np.argsort(xval)[idx].astype(idx_dtype)
# due to uniqueness, we expect indices same
assert np.all(xval[np.sort(yval)] == xval[np.sort(goal)])
utt.assert_allclose(xval[np.sort(goal)], xval[np.sort(yval)])
@utt.parameterized.expand(chain(
product(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论