提交 edccf958 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

added some cases and formatting to the test

上级 0aee4085
...@@ -201,7 +201,10 @@ def test_max(): ...@@ -201,7 +201,10 @@ def test_max():
gpu_pattern = tuple(gpu_pattern) gpu_pattern = tuple(gpu_pattern)
return gpu_pattern return gpu_pattern
for shape, pattern in [((100,3,1300),[1]), for shape, pattern in [((1,1),(1,)),
((2,1),(1,)),
((1,2),(1,)),
((100,3,1300),[1]),
((0,),[0]),((5,),[0]), ((0,),[0]),((5,),[0]),
((0,0),[0,1]),((1,0),[0,1]),((5,4),[0,1]),((33,31),[0,1]),((5,4),[1]),((5,4),[0]),#need something bigger then 32 for some opt test. ((0,0),[0,1]),((1,0),[0,1]),((5,4),[0,1]),((33,31),[0,1]),((5,4),[1]),((5,4),[0]),#need something bigger then 32 for some opt test.
((5,4,3),[0]),((5,4,3),[1]),((5,4,3),[0,1]),((5,4,3),[2]),((5,4,3),[1,2]),((5,4,3),[0,1,2]), ((5,4,3),[0]),((5,4,3),[1]),((5,4,3),[0,1]),((5,4,3),[2]),((5,4,3),[1,2]),((5,4,3),[0,1,2]),
...@@ -270,10 +273,19 @@ def test_max(): ...@@ -270,10 +273,19 @@ def test_max():
#example in debug mode with unittests.rseed=9275 #example in debug mode with unittests.rseed=9275
orig_rtol = theano.tensor.basic.float32_rtol orig_rtol = theano.tensor.basic.float32_rtol
theano.tensor.basic.float32_rtol = 2e-5 theano.tensor.basic.float32_rtol = 2e-5
assert _allclose(f2(val), f(val)), ('shape', shape, f2_val = f2(val)
'pattern', pattern, f_val = f(val)
sum([shape[i] for i in pattern]), if not _allclose(f2_val, f_val):
f2(val), f(val), val) print 'failed for the following arguments: '
print 'shape:',shape
print 'pattern: ',pattern
print 'input:'
print val
print 'correct output: '
print f2_val
print 'actual output: '
print f_val
assert False
finally: finally:
theano.tensor.basic.float32_rtol = orig_rtol theano.tensor.basic.float32_rtol = orig_rtol
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论