提交 f92772ed authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix tests for error by activating the flag

上级 daedf052
...@@ -7,6 +7,7 @@ from nose.tools import assert_raises, assert_true ...@@ -7,6 +7,7 @@ from nose.tools import assert_raises, assert_true
import theano import theano
from theano import tensor from theano import tensor
from theano.configparser import change_flags
from theano.gof.opt import check_stack_trace from theano.gof.opt import check_stack_trace
from theano.tests import unittest_tools as utt from theano.tests import unittest_tools as utt
from theano.tensor.nnet import (corr, corr3d, conv2d_transpose, from theano.tensor.nnet import (corr, corr3d, conv2d_transpose,
...@@ -229,6 +230,7 @@ class TestAssertConvShape(unittest.TestCase): ...@@ -229,6 +230,7 @@ class TestAssertConvShape(unittest.TestCase):
class TestAssertShape(unittest.TestCase): class TestAssertShape(unittest.TestCase):
@change_flags([("conv.assert_shape",True)])
def test_basic(self): def test_basic(self):
x = tensor.tensor4() x = tensor.tensor4()
s1 = tensor.iscalar() s1 = tensor.iscalar()
...@@ -244,6 +246,7 @@ class TestAssertShape(unittest.TestCase): ...@@ -244,6 +246,7 @@ class TestAssertShape(unittest.TestCase):
assert_raises(AssertionError, f, v, 0, 7) assert_raises(AssertionError, f, v, 0, 7)
assert_raises(AssertionError, f, v, 7, 7) assert_raises(AssertionError, f, v, 7, 7)
@change_flags([("conv.assert_shape",True)])
def test_shape_check_conv2d(self): def test_shape_check_conv2d(self):
input = tensor.tensor4() input = tensor.tensor4()
filters = tensor.tensor4() filters = tensor.tensor4()
...@@ -261,6 +264,7 @@ class TestAssertShape(unittest.TestCase): ...@@ -261,6 +264,7 @@ class TestAssertShape(unittest.TestCase):
numpy.zeros((3, 5, 7, 11), dtype='float32'), numpy.zeros((3, 5, 7, 11), dtype='float32'),
numpy.zeros((7, 5, 2, 2), dtype='float32')) numpy.zeros((7, 5, 2, 2), dtype='float32'))
@change_flags([("conv.assert_shape",True)])
def test_shape_check_conv3d(self): def test_shape_check_conv3d(self):
input = tensor.tensor5() input = tensor.tensor5()
filters = tensor.tensor5() filters = tensor.tensor5()
...@@ -278,6 +282,7 @@ class TestAssertShape(unittest.TestCase): ...@@ -278,6 +282,7 @@ class TestAssertShape(unittest.TestCase):
numpy.zeros((3, 5, 7, 11, 13), dtype='float32'), numpy.zeros((3, 5, 7, 11, 13), dtype='float32'),
numpy.zeros((7, 5, 2, 2, 2), dtype='float32')) numpy.zeros((7, 5, 2, 2, 2), dtype='float32'))
@change_flags([("conv.assert_shape",True)])
def test_shape_check_conv2d_grad_wrt_inputs(self): def test_shape_check_conv2d_grad_wrt_inputs(self):
output_grad = tensor.tensor4() output_grad = tensor.tensor4()
filters = tensor.tensor4() filters = tensor.tensor4()
...@@ -291,6 +296,7 @@ class TestAssertShape(unittest.TestCase): ...@@ -291,6 +296,7 @@ class TestAssertShape(unittest.TestCase):
numpy.zeros((3, 6, 5, 9), dtype='float32'), numpy.zeros((3, 6, 5, 9), dtype='float32'),
numpy.zeros((7, 6, 3, 3), dtype='float32')) numpy.zeros((7, 6, 3, 3), dtype='float32'))
@change_flags([("conv.assert_shape",True)])
def test_shape_check_conv3d_grad_wrt_inputs(self): def test_shape_check_conv3d_grad_wrt_inputs(self):
output_grad = tensor.tensor5() output_grad = tensor.tensor5()
filters = tensor.tensor5() filters = tensor.tensor5()
...@@ -304,6 +310,7 @@ class TestAssertShape(unittest.TestCase): ...@@ -304,6 +310,7 @@ class TestAssertShape(unittest.TestCase):
numpy.zeros((3, 6, 5, 9, 11), dtype='float32'), numpy.zeros((3, 6, 5, 9, 11), dtype='float32'),
numpy.zeros((7, 6, 3, 3, 3), dtype='float32')) numpy.zeros((7, 6, 3, 3, 3), dtype='float32'))
@change_flags([("conv.assert_shape",True)])
def test_shape_check_conv2d_grad_wrt_weights(self): def test_shape_check_conv2d_grad_wrt_weights(self):
input = tensor.tensor4() input = tensor.tensor4()
output_grad = tensor.tensor4() output_grad = tensor.tensor4()
...@@ -317,6 +324,7 @@ class TestAssertShape(unittest.TestCase): ...@@ -317,6 +324,7 @@ class TestAssertShape(unittest.TestCase):
numpy.zeros((3, 6, 7, 11), dtype='float32'), numpy.zeros((3, 6, 7, 11), dtype='float32'),
numpy.zeros((3, 7, 5, 9), dtype='float32')) numpy.zeros((3, 7, 5, 9), dtype='float32'))
@change_flags([("conv.assert_shape",True)])
def test_shape_check_conv3d_grad_wrt_weights(self): def test_shape_check_conv3d_grad_wrt_weights(self):
input = tensor.tensor5() input = tensor.tensor5()
output_grad = tensor.tensor5() output_grad = tensor.tensor5()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论