提交 91d149ab authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Fix flaky RNG usage in tests.tensor.test_math.test_cov

上级 cae7c1a5
...@@ -2557,6 +2557,8 @@ def test_norm(): ...@@ -2557,6 +2557,8 @@ def test_norm():
def test_cov(): def test_cov():
rng = np.random.default_rng(utt.fetch_seed())
x = matrix("x") x = matrix("x")
y = matrix("y") y = matrix("y")
...@@ -2564,14 +2566,14 @@ def test_cov(): ...@@ -2564,14 +2566,14 @@ def test_cov():
c = cov(x, rowvar=rowvar, bias=bias, ddof=ddof) c = cov(x, rowvar=rowvar, bias=bias, ddof=ddof)
f = function([x], c) f = function([x], c)
data = np.asarray(np.random.random((3, 5)), dtype=config.floatX) data = np.asarray(rng.random((3, 5)), dtype=config.floatX)
assert np.allclose(f(data), np.cov(data, rowvar=rowvar, bias=bias, ddof=ddof)) assert np.allclose(f(data), np.cov(data, rowvar=rowvar, bias=bias, ddof=ddof))
c = cov(x, y=y, rowvar=rowvar, bias=bias, ddof=ddof) c = cov(x, y=y, rowvar=rowvar, bias=bias, ddof=ddof)
f = function([x, y], c) f = function([x, y], c)
data = np.asarray(np.random.random((3, 5)), dtype=config.floatX) data = np.asarray(rng.random((3, 5)), dtype=config.floatX)
y_val = np.asarray(np.random.random((3, 5)), dtype=config.floatX) y_val = np.asarray(rng.random((3, 5)), dtype=config.floatX)
assert np.allclose( assert np.allclose(
f(data, y_val), np.cov(data, rowvar=rowvar, y=y_val, bias=bias, ddof=ddof) f(data, y_val), np.cov(data, rowvar=rowvar, y=y_val, bias=bias, ddof=ddof)
) )
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论