提交 af9d4e8b authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Avoid printing of an error in one test when it is expected.

上级 6b4e56d4
...@@ -385,8 +385,13 @@ class TestEquilibrium(object): ...@@ -385,8 +385,13 @@ class TestEquilibrium(object):
e = op3(op4(x, y)) e = op3(op4(x, y))
g = Env([x, y, z], [e]) g = Env([x, y, z], [e])
print 'before', g print 'before', g
# display pesky warnings along with stdout
oldstderr = sys.stderr oldstderr = sys.stderr
sys.stderr = sys.stdout # display pesky warnings along with stdout sys.stderr = sys.stdout
# also silence logger for 'theano.gof.opt'
_logger = logging.getLogger('theano.gof.opt')
oldlevel = _logger.getEffectiveLevel()
_logger.setLevel(logging.CRITICAL)
try: try:
opt = EquilibriumOptimizer( opt = EquilibriumOptimizer(
[PatternSub((op1, 'x', 'y'), (op2, 'x', 'y')), [PatternSub((op1, 'x', 'y'), (op2, 'x', 'y')),
...@@ -397,6 +402,7 @@ class TestEquilibrium(object): ...@@ -397,6 +402,7 @@ class TestEquilibrium(object):
opt.optimize(g) opt.optimize(g)
finally: finally:
sys.stderr = oldstderr sys.stderr = oldstderr
_logger.setLevel(oldlevel)
print 'after', g print 'after', g
assert str(g) == '[Op4(x, y)]' assert str(g) == '[Op4(x, y)]'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论