提交 b7136d80 authored 作者: affanv14's avatar affanv14

add test to check default values

上级 594dbd74
...@@ -579,6 +579,20 @@ class T_function(unittest.TestCase): ...@@ -579,6 +579,20 @@ class T_function(unittest.TestCase):
if not isinstance(key, theano.gof.Constant): if not isinstance(key, theano.gof.Constant):
assert (val[0] is None) assert (val[0] is None)
def test_default_values(self):
"""
Check that default values are restored
when an exception occurs in interactive mode.
"""
a, b = T.dscalars('a', 'b')
c = a + b
func = theano.function([theano.In(a, name='first'), theano.In(b, value=1, name='second')], c)
x = func(first=1)
try:
func(second=2)
except TypeError:
assert(func(first=1) == x)
class T_picklefunction(unittest.TestCase): class T_picklefunction(unittest.TestCase):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论