提交 ba320eeb authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Small refactoring of tests.

上级 dcfe260e
...@@ -699,7 +699,8 @@ def test_dot_mv(): ...@@ -699,7 +699,8 @@ def test_dot_mv():
assert sum([isinstance(node.op, T.Dot) for node in assert sum([isinstance(node.op, T.Dot) for node in
f.maker.env.toposort() ]) == 1 f.maker.env.toposort() ]) == 1
def test_gemv1(): class TestGemv(TestCase):
def test_gemv1(self):
''' test vector1+dot(matrix,vector2) ''' ''' test vector1+dot(matrix,vector2) '''
rng = numpy.random.RandomState(unittest_tools.fetch_seed()) rng = numpy.random.RandomState(unittest_tools.fetch_seed())
v1 = theano.shared(numpy.array(rng.uniform(size=(2,)), dtype='float32')) v1 = theano.shared(numpy.array(rng.uniform(size=(2,)), dtype='float32'))
...@@ -731,7 +732,7 @@ def test_gemv1(): ...@@ -731,7 +732,7 @@ def test_gemv1():
if config.mode != 'FAST_COMPILE': if config.mode != 'FAST_COMPILE':
assert topo[0].op.inplace==True assert topo[0].op.inplace==True
def test_gemv2(): def test_gemv2(self):
''' test vector1+dot(vector2,matrix) ''' ''' test vector1+dot(vector2,matrix) '''
rng = numpy.random.RandomState(unittest_tools.fetch_seed()) rng = numpy.random.RandomState(unittest_tools.fetch_seed())
v1 = theano.shared(numpy.array(rng.uniform(size=(2,)), dtype='float32')) v1 = theano.shared(numpy.array(rng.uniform(size=(2,)), dtype='float32'))
...@@ -761,7 +762,6 @@ def test_gemv2(): ...@@ -761,7 +762,6 @@ def test_gemv2():
if config.mode != 'FAST_COMPILE': if config.mode != 'FAST_COMPILE':
assert topo[-1].op.inplace==True assert topo[-1].op.inplace==True
class TestGemv(TestCase):
def test_gemv_dimensions(self): def test_gemv_dimensions(self):
A = T.matrix('A') A = T.matrix('A')
x, y = T.vectors('x', 'y') x, y = T.vectors('x', 'y')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论