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

Catch zero div exception in test

上级 8575d2d2
......@@ -363,7 +363,8 @@ class T_Images2Neibs(unittest_tools.InferShapeTester):
self.assertRaises(ValueError, f, neibs,
(1, 1, 3, 320, 320))
# End up with a step of 0
self.assertRaises(ValueError, f, neibs,
# This can lead to division by zero in DebugMode
self.assertRaises((ValueError, ZeroDivisionError), f, neibs,
(3, 320, 320, 1))
def speed_neibs(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论