提交 c1a41aa0 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

changed neighbours test to work with new TypeError system

上级 1ce5cffc
...@@ -260,7 +260,7 @@ class T_Images2Neibs(unittest_tools.InferShapeTester): ...@@ -260,7 +260,7 @@ class T_Images2Neibs(unittest_tools.InferShapeTester):
def fn(images): def fn(images):
return images2neibs(images, (3, 3), mode='wrap_centered') return images2neibs(images, (3, 3), mode='wrap_centered')
self.assertRaises(NotImplementedError, unittest_tools.verify_grad, self.assertRaises(TypeError, unittest_tools.verify_grad,
fn, [images_val], mode=self.mode) fn, [images_val], mode=self.mode)
...@@ -278,7 +278,7 @@ class T_Images2Neibs(unittest_tools.InferShapeTester): ...@@ -278,7 +278,7 @@ class T_Images2Neibs(unittest_tools.InferShapeTester):
# are not the same. # are not the same.
def fn(images): def fn(images):
return images2neibs(images, (2, 2), (1, 1)) return images2neibs(images, (2, 2), (1, 1))
self.assertRaises(NotImplementedError, self.assertRaises(TypeError,
unittest_tools.verify_grad, fn, [images_val], unittest_tools.verify_grad, fn, [images_val],
mode=self.mode) mode=self.mode)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论