提交 e3dad04e authored 作者: James Bergstra's avatar James Bergstra

added some tests for Add

上级 bbe27f18
...@@ -886,28 +886,28 @@ class T_Stack(unittest.TestCase): ...@@ -886,28 +886,28 @@ class T_Stack(unittest.TestCase):
# class T_add(unittest.TestCase): class T_add(unittest.TestCase):
# def test_complex_all_ops(self): def test_complex_all_ops(self):
# for nbits in (64, 128): for nbits in (64, 128):
# a = as_tensor(numpy.ones(3, dtype='complex%i' % nbits)+0.5j) a = value(numpy.ones(3, dtype='complex%i' % nbits)+0.5j)
# b = as_tensor(numpy.ones(3, dtype='complex%i' % nbits)+1.5j) b = value(numpy.ones(3, dtype='complex%i' % nbits)+1.5j)
# tests = (("+", lambda x,y: x+y), tests = (("+", lambda x,y: x+y),
# ("-", lambda x,y: x-y), ("-", lambda x,y: x-y),
# ("*", lambda x,y: x*y), ("*", lambda x,y: x*y),
# ("/", lambda x,y: x/y)) ("/", lambda x,y: x/y))
# for s, fn in tests: for s, fn in tests:
# f = Function([a,b], [fn(a, b)], linker_cls = gof.CLinker) f = function([a,b], [fn(a, b)], linker = gof.CLinker)
# self.failUnless(numpy.all(fn(a.data, b.data) == f(a.data, b.data))) self.failUnless(numpy.all(fn(a.data, b.data) == f(a.data, b.data)))
# def test_grad_scalar_l(self): def test_grad_scalar_l(self):
# verify_grad(self, Add, [numpy.asarray([3.0]), numpy.random.rand(3)]) verify_grad(self, add, [numpy.asarray([3.0]), numpy.random.rand(3)])
# def test_grad_scalar_r(self): def test_grad_scalar_r(self):
# verify_grad(self, Add, [numpy.random.rand(3), numpy.asarray([3.0])]) verify_grad(self, add, [numpy.random.rand(3), numpy.asarray([3.0])])
# def test_grad_row(self): def test_grad_row(self):
# verify_grad(self, Add, [numpy.random.rand(3, 5), numpy.random.rand(1, 5)]) verify_grad(self, add, [numpy.random.rand(3, 5), numpy.random.rand(1, 5)])
# def test_grad_col(self): def test_grad_col(self):
# verify_grad(self, Add, [numpy.random.rand(3, 5), numpy.random.rand(3, 1)]) verify_grad(self, add, [numpy.random.rand(3, 5), numpy.random.rand(3, 1)])
# class T_abs(unittest.TestCase): # class T_abs(unittest.TestCase):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论