提交 6873a47e authored 作者: David Warde-Farley's avatar David Warde-Farley

Turn the CholeskyGrad test into a generator.

The three calls are verifying slightly different things.
上级 7d0addfa
...@@ -69,11 +69,11 @@ def test_cholesky_grad(): ...@@ -69,11 +69,11 @@ def test_cholesky_grad():
r = rng.randn(5, 5) r = rng.randn(5, 5)
pd = numpy.dot(r, r.T) pd = numpy.dot(r, r.T)
# Check the default. # Check the default.
utt.verify_grad(cholesky, [pd], rng=rng) yield utt.verify_grad, cholesky, [pd], 3, rng
# Explicit lower-triangular. # Explicit lower-triangular.
utt.verify_grad(Cholesky(lower=True), [pd], rng=rng) yield utt.verify_grad, Cholesky(lower=True), [pd], 3, rng
# Explicit upper-triangular. # Explicit upper-triangular.
utt.verify_grad(Cholesky(lower=False), [pd], rng=rng) yield utt.verify_grad, Cholesky(lower=False), [pd], 3, rng
def test_cholesky_shape(): def test_cholesky_shape():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论