提交 50dac4fa authored 作者: James Bergstra's avatar James Bergstra

added tests for tensor eq, neq

上级 07b8cb31
...@@ -957,6 +957,22 @@ class _test_comparison(unittest.TestCase): ...@@ -957,6 +957,22 @@ class _test_comparison(unittest.TestCase):
v = fn(l, r) v = fn(l, r)
self.failUnless(numpy.all(v == (l >= r)), (v, (l>=r))) self.failUnless(numpy.all(v == (l >= r)), (v, (l>=r)))
def test_eq(self):
x, y = fvector(), fvector()
fn = function([x,y], [eq(x,y)])
l = numpy.asarray([0.,-1.,1.])
r = numpy.asarray([0.,1.,-1.])
v = fn(l, r)
self.failUnless(numpy.all(v == (l == r)), (v, (l==r)))
def test_neq(self):
x, y = fvector(), fvector()
fn = function([x,y], [neq(x, y)])
l = numpy.asarray([0.,-1.,1.])
r = numpy.asarray([0.,1.,-1.])
v = fn(l, r)
self.failUnless(numpy.all(v == (l != r)), (v, (l!=r)))
class _test_bitwise(unittest.TestCase): class _test_bitwise(unittest.TestCase):
def test_or(self): def test_or(self):
x, y = bvector(), bvector() x, y = bvector(), bvector()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论