提交 45e43f1a authored 作者: Matthew Rocklin's avatar Matthew Rocklin

skip sympy tests with skiptest

上级 fa5a65c0
...@@ -7,12 +7,12 @@ try: ...@@ -7,12 +7,12 @@ try:
xs = sympy.Symbol('x') xs = sympy.Symbol('x')
ys = sympy.Symbol('y') ys = sympy.Symbol('y')
except ImportError: except ImportError:
sympy = False from nose.plugins.skip import SkipTest
raise skiptest('optional package sympy disabled')
xt, yt = floats('xy') xt, yt = floats('xy')
def test_SymPyCCode(): def test_SymPyCCode():
if not sympy: return
op = SymPyCCode([xs, ys], xs + ys) op = SymPyCCode([xs, ys], xs + ys)
e = op(xt, yt) e = op(xt, yt)
g = theano.gof.FunctionGraph([xt, yt], [e]) g = theano.gof.FunctionGraph([xt, yt], [e])
...@@ -20,14 +20,12 @@ def test_SymPyCCode(): ...@@ -20,14 +20,12 @@ def test_SymPyCCode():
assert fn(1.0, 2.0) == 3.0 assert fn(1.0, 2.0) == 3.0
def test_grad(): def test_grad():
if not sympy: return
op = SymPyCCode([xs], xs**2) op = SymPyCCode([xs], xs**2)
zt = op(xt) zt = op(xt)
ztprime = theano.grad(zt, xt) ztprime = theano.grad(zt, xt)
assert ztprime.owner.op.expr == 2*xs assert ztprime.owner.op.expr == 2*xs
def test_multivar_grad(): def test_multivar_grad():
if not sympy: return
op = SymPyCCode([xs, ys], xs**2 + ys**2) op = SymPyCCode([xs, ys], xs**2 + ys**2)
zt = op(xt, yt) zt = op(xt, yt)
dzdx, dzdy = theano.grad(zt, [xt, yt]) dzdx, dzdy = theano.grad(zt, [xt, yt])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论