提交 fdaa40ba authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Clarify assert statement in test_get_vector_length

上级 0706d9ad
...@@ -1019,7 +1019,7 @@ def test_get_vector_length(): ...@@ -1019,7 +1019,7 @@ def test_get_vector_length():
mode = aesara.compile.get_default_mode().excluding("constant_folding") mode = aesara.compile.get_default_mode().excluding("constant_folding")
f = function([x, y], [b, c, a], mode=mode) f = function([x, y], [b, c, a], mode=mode)
topo = f.maker.fgraph.toposort() topo = f.maker.fgraph.toposort()
assert [True for node in topo if isinstance(node.op, MakeVector)] assert any([True for node in topo if isinstance(node.op, MakeVector)])
assert np.allclose(f(4, 5), [5, 9, 4]) assert np.allclose(f(4, 5), [5, 9, 4])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论