提交 a9dbce47 authored 作者: Frederic's avatar Frederic

Better test error message.

上级 810e5197
#from nose.plugins.skip import SkipTest #from nose.plugins.skip import SkipTest
#import traceback #import traceback
from copy import copy
import itertools import itertools
import sys import sys
...@@ -24,7 +25,6 @@ from theano.tensor.blas import (_dot22, _dot22scalar, res_is_a, _as_scalar, ...@@ -24,7 +25,6 @@ from theano.tensor.blas import (_dot22, _dot22scalar, res_is_a, _as_scalar,
InconsistencyError, Ger, ger, ger_destructive) InconsistencyError, Ger, ger, ger_destructive)
from unittest import TestCase from unittest import TestCase
from theano.tests import unittest_tools from theano.tests import unittest_tools
from copy import copy, deepcopy
from theano import Param, shared, config from theano import Param, shared, config
from test_basic import (_approx_eq, as_tensor_variable, inplace_func, from test_basic import (_approx_eq, as_tensor_variable, inplace_func,
...@@ -361,11 +361,8 @@ class t_gemm(TestCase): ...@@ -361,11 +361,8 @@ class t_gemm(TestCase):
z = tz.get_value(borrow=True, return_internal_type=True) z = tz.get_value(borrow=True, return_internal_type=True)
z[:, :, i] = z_i z[:, :, i] = z_i
self.assertTrue( unittest_tools.assert_allclose(z_after[:, :, i],
_approx_eq(z_after[:, :, i], tz.get_value(borrow=True)[:, :, i])
tz.get_value(borrow=True)[:, :, i]),
(z_orig[:, :, i], z_after[:, :, i],
z[:, :, i], z_after[:, :, i] - z[:, :, i]))
tz_i = gemm_no_inplace(tz[:, :, i], ta, tx[ tz_i = gemm_no_inplace(tz[:, :, i], ta, tx[
:, :, i], ty[:, :, i], tb) :, :, i], ty[:, :, i], tb)
...@@ -374,11 +371,8 @@ class t_gemm(TestCase): ...@@ -374,11 +371,8 @@ class t_gemm(TestCase):
mode=compile.Mode(optimizer=None, linker=l)) mode=compile.Mode(optimizer=None, linker=l))
for j in xrange(3): for j in xrange(3):
g_i() g_i()
self.assertTrue( unittest_tools.assert_allclose(z_after[:, :, i],
_approx_eq(z_after[:, :, i], tz.get_value(borrow=True)[:, :, i])
tz.get_value(borrow=True)[:, :, i]),
(z_orig[:, :, i], z_after[:, :, i],
z[:, :, i], z_after[:, :, i] - z[:, :, i]))
t(C, A, B) t(C, A, B)
t(C.transpose((1, 0, 2)), A, B) t(C.transpose((1, 0, 2)), A, B)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论