提交 23e0ad2e authored 作者: Frederic Bastien's avatar Frederic Bastien

Add information in assertion error.

上级 2c292408
...@@ -1776,14 +1776,13 @@ def test_local_useless_subtensor(): ...@@ -1776,14 +1776,13 @@ def test_local_useless_subtensor():
((slice(0, 3), slice(0, 13)), True), ((slice(0, 3), slice(0, 13)), True),
((slice(0, 3), slice(0, 2)), False), ((slice(0, 3), slice(0, 2)), False),
((slice(0, 1), slice(0, None)), False), ((slice(0, 1), slice(0, None)), False),
((slice(0, 1), 1), False), ((slice(0, 1), 1), False)]:
]:
f = function([x], tensor.exp(x_c).__getitem__(dims), mode=mode_opt) f = function([x], tensor.exp(x_c).__getitem__(dims), mode=mode_opt)
# theano.printing.debugprint(f) # theano.printing.debugprint(f)
prog = f.maker.fgraph.toposort() prog = f.maker.fgraph.toposort()
if res: if res:
assert isinstance(prog[0].op, theano.tensor.SpecifyShape), dims assert isinstance(prog[0].op, theano.tensor.SpecifyShape), dims
assert prog[1].op == tensor.exp, dims assert prog[1].op == tensor.exp, (dims, prog)
assert len(prog) == 2, dims assert len(prog) == 2, dims
else: else:
assert any([isinstance(node.op, Subtensor) for node in prog]) assert any([isinstance(node.op, Subtensor) for node in prog])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论