提交 8f8b6e1f authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

add 1D test for nonzero

上级 46bfeaf6
......@@ -1904,9 +1904,16 @@ def test_nonzero():
f = function([m_symb], nonzero(m_symb))
result = f(m)
if not isinstance(result, list):
result = [result]
for i, j in zip(result, numpy.nonzero(m)):
assert numpy.allclose(i, j)
rand1d = rand(8)
rand1d[rand1d > rand1d.mean()] = 0
check(rand1d)
rand2d = rand(8, 9)
rand2d[rand2d > rand2d.mean()] = 0
check(rand2d)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论