提交 ebe0a6c1 authored 作者: Frederic's avatar Frederic

Add assert in tests.

上级 697d1f24
......@@ -3550,6 +3550,8 @@ class T_Join_and_Split(unittest.TestCase):
m = self.shared(rng.rand(4, 6).astype(self.floatX))
o = self.split_op(2)(m, 0, [4, 0])
f = function([], o, mode=self.mode)
assert any([isinstance(node.op, self.split_op)
for node in f.maker.fgraph.toposort()])
o1, o2 = f()
assert numpy.allclose(o1, m.get_value(borrow=True))
assert numpy.allclose(o2, m.get_value(borrow=True)[4:])
......@@ -3559,6 +3561,8 @@ class T_Join_and_Split(unittest.TestCase):
m = self.shared(rng.rand(4, 6).astype(self.floatX))
o = self.split_op(2)(m, 0, [5, -1])
f = function([], o, mode=self.mode)
assert any([isinstance(node.op, self.split_op)
for node in f.maker.fgraph.toposort()])
self.assertRaises(ValueError, f)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论