提交 22d0a2a4 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Update test to have more interesting cases tested.

上级 a7be8923
...@@ -1807,13 +1807,15 @@ class test_local_subtensor_merge(unittest.TestCase): ...@@ -1807,13 +1807,15 @@ class test_local_subtensor_merge(unittest.TestCase):
#print topo[-1].op #print topo[-1].op
assert isinstance(topo[-1].op, theano.compile.function_module.DeepCopyOp) assert isinstance(topo[-1].op, theano.compile.function_module.DeepCopyOp)
b_r = self.rng.permutation(range(-8,8))[:2] b_r = self.rng.permutation(range(-4,4))[:3]
e_r = self.rng.permutation(range(-8,8))[:2] e_r = self.rng.permutation(range(-4,4))[:3]
i_r = self.rng.permutation(range(-8,8))[:2] i_r = self.rng.permutation(range(-4,4))[:3]
s_r = self.rng.permutation([-7,-6,-5,-4,-3,-2,-1,1,2,3,4,5,6,7])[:2] s_r = self.rng.permutation([-3,-2,-1,1,2,3])[:3]
for x_s in self.x_shapes: for x_s in self.x_shapes:
n_index_err = 0
n_ok = 0
x_val = self.rng.uniform(size=x_s).astype(config.floatX) x_val = self.rng.uniform(size=x_s).astype(config.floatX)
for b_v in b_r: for b_v in b_r:
for e_v in e_r: for e_v in e_r:
...@@ -1825,13 +1827,18 @@ class test_local_subtensor_merge(unittest.TestCase): ...@@ -1825,13 +1827,18 @@ class test_local_subtensor_merge(unittest.TestCase):
try: try:
x_val[b_v:e_v:s_v][i_v] x_val[b_v:e_v:s_v][i_v]
except IndexError: except IndexError:
n_index_err += 1
self.assertRaises(IndexError, self.assertRaises(IndexError,
f, x_val, b_v, e_v, s_v, i_v) f, x_val, b_v, e_v, s_v, i_v)
else: else:
# Executed if the "try" clause did not # Executed if the "try" clause did not raise
# raise an exception # any exception
n_ok += 1
f(x_val, b_v, e_v, s_v, i_v) f(x_val, b_v, e_v, s_v, i_v)
print 'shape: %s' % (x_s,)
print '%% OK: %f' % (float(n_ok) * 100 / (n_ok + n_index_err))
class Test_alloc_zero(unittest.TestCase): class Test_alloc_zero(unittest.TestCase):
def setUp(self): def setUp(self):
mode = theano.compile.mode.get_default_mode() mode = theano.compile.mode.get_default_mode()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论