提交 57a2f9c8 authored 作者: Ramana.S's avatar Ramana.S

Slight improvement made to the test

上级 c4fb7e56
...@@ -40,17 +40,18 @@ class OpDecoratorTests(utt.InferShapeTester): ...@@ -40,17 +40,18 @@ class OpDecoratorTests(utt.InferShapeTester):
y = dvector('y') y = dvector('y')
y.tag.test_value = [0, 0] y.tag.test_value = [0, 0]
with self.assertRaises(NotImplementedError): with self.assertRaisesRegexp(NotImplementedError, "itypes not defined") :
@as_op(itypes=[dmatrix, dvector], otypes=[])
def none_otypes(x,y):
return np.dot(x,y)
@as_op(itypes=[], otypes=dvector) @as_op(itypes=[], otypes=dvector)
def none_itypes(x,y): def none_itypes(x,y):
return np.dot(x,y) return np.dot(x,y)
none_itypes(x,y) none_itypes(x,y)
with self.assertRaisesRegexp(NotImplementedError, "otypes not defined") :
@as_op(itypes=[dmatrix, dvector], otypes=[])
def none_otypes(x,y):
return np.dot(x,y)
none_otypes(x, y) none_otypes(x, y)
def test_2arg(self): def test_2arg(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论