提交 3bd247e4 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Avoid warnings in tests.tensor.test_var.test_numpy_method

上级 40f6d7db
...@@ -12,6 +12,7 @@ from aesara.tensor.type import ( ...@@ -12,6 +12,7 @@ from aesara.tensor.type import (
TensorType, TensorType,
cscalar, cscalar,
dmatrix, dmatrix,
dscalar,
dvector, dvector,
iscalar, iscalar,
ivector, ivector,
...@@ -23,41 +24,37 @@ from aesara.tensor.var import TensorConstant ...@@ -23,41 +24,37 @@ from aesara.tensor.var import TensorConstant
@pytest.mark.parametrize( @pytest.mark.parametrize(
"fct", "fct, value",
[ [
np.arccos, (np.arccos, 0.5),
np.arccosh, (np.arccosh, 1.0),
np.arcsin, (np.arcsin, 0.5),
np.arcsinh, (np.arcsinh, 0.5),
np.arctan, (np.arctan, 0.5),
np.arctanh, (np.arctanh, 0.5),
# np.ceil, np.floor, np.trunc, (np.cos, 0.5),
np.cos, (np.cosh, 0.5),
np.cosh, (np.deg2rad, 0.5),
np.deg2rad, (np.exp, 0.5),
np.exp, (np.exp2, 0.5),
np.exp2, (np.expm1, 0.5),
np.expm1, (np.log, 0.5),
np.log, (np.log10, 0.5),
np.log10, (np.log1p, 0.5),
np.log1p, (np.log2, 0.5),
np.log2, (np.rad2deg, 0.5),
np.rad2deg, (np.sin, 0.5),
np.sin, (np.sinh, 0.5),
np.sinh, (np.sqrt, 0.5),
np.sqrt, (np.tan, 0.5),
np.tan, (np.tanh, 0.5),
np.tanh,
], ],
) )
def test_numpy_method(fct): def test_numpy_method(fct, value):
# This type of code is used frequently by PyMC3 users x = dscalar("x")
x = dmatrix("x")
data = np.random.rand(5, 5)
x.tag.test_value = data
y = fct(x) y = fct(x)
f = aesara.function([x], y) f = aesara.function([x], y)
utt.assert_allclose(np.nan_to_num(f(data)), np.nan_to_num(fct(data))) utt.assert_allclose(np.nan_to_num(f(value)), np.nan_to_num(fct(value)))
def test_infix_dot_method(): def test_infix_dot_method():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论