提交 501dc901 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

moving tensor testers

上级 970f985d
......@@ -15,7 +15,7 @@ def Env(i, o):
e = gof.Env(i, o)
return e
class _test_DimShuffle(unittest.TestCase):
class test_DimShuffle(unittest.TestCase):
def with_linker(self, linker):
for xsh, shuffle, zsh in [((2, 3), (1, 'x', 0), (3, 1, 2)),
......@@ -35,7 +35,7 @@ class _test_DimShuffle(unittest.TestCase):
self.with_linker(gof.PerformLinker())
class _test_Broadcast(unittest.TestCase):
class test_Broadcast(unittest.TestCase):
def with_linker(self, linker):
for xsh, ysh in [((3, 5), (3, 5)),
......@@ -119,7 +119,7 @@ class _test_Broadcast(unittest.TestCase):
assert (f(xv) == zv).all()
class _test_CAReduce(unittest.TestCase):
class test_CAReduce(unittest.TestCase):
def with_linker(self, linker):
for xsh, tosum in [((5, 6), None),
......
......@@ -61,7 +61,7 @@ def inputs(xbc = (0, 0), ybc = (0, 0), zbc = (0, 0)):
ds = lambda x, y: DimShuffle(x.type.broadcastable, y)(x)
class _test_dimshuffle_lift(unittest.TestCase):
class test_dimshuffle_lift(unittest.TestCase):
def test_double_transpose(self):
x, y, z = inputs()
......@@ -102,7 +102,7 @@ from theano.tensor import *
#from sandbox import pprint
class _test_greedy_distribute(unittest.TestCase):
class test_greedy_distribute(unittest.TestCase):
def test_main(self):
a, b, c, d, x, y, z = matrices('abcdxyz')
e = (a/z + b/x) * x * z
......@@ -115,7 +115,7 @@ class _test_greedy_distribute(unittest.TestCase):
class _test_canonize(unittest.TestCase):
class test_canonize(unittest.TestCase):
def test_muldiv(self):
x, y, z = matrices('xyz')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论