提交 3d9ec506 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

fixed test

上级 8cb589f1
...@@ -139,6 +139,8 @@ class TestBinCountOp(utt.InferShapeTester): ...@@ -139,6 +139,8 @@ class TestBinCountOp(utt.InferShapeTester):
f4 = theano.function([x], bincount(x, minlength=5)) f4 = theano.function([x], bincount(x, minlength=5))
assert (np.bincount(a, minlength=23) == f3(a)).all() assert (np.bincount(a, minlength=23) == f3(a)).all()
assert (np.bincount(a, minlength=5) == f4(a)).all() assert (np.bincount(a, minlength=5) == f4(a)).all()
# skip the following test when using unsigned ints
if not dtype.startswith('u'):
a[0] = -1 a[0] = -1
f5 = theano.function([x], bincount(x, assert_nonneg=True)) f5 = theano.function([x], bincount(x, assert_nonneg=True))
self.assertRaises(AssertionError, f5, a) self.assertRaises(AssertionError, f5, a)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论